Quel modèle garder pour faire des prédictions dans le cas d'une validation k-fold ?

Question

Avec une évaluation k-fold, on se retrouve avec 1 évaluation mais k modèles. Quel modèle faut-il utiliser pour faire des prédictions sur de nouvelles observations ?

Answer 1

Il n’y a pas une évaluation mais k-évaluation que l’on va moyenner. Le K-fold sert à choisir les hyperparamètres et non pas le modèle. Dès qu’on a évalué les meilleurs hyperparamètres, il faut ré-entraîné un modèle avec les hyperparamètres choisis et sur tout le dataset d'entraînement et de validation (pas le dataset test attention).