Construis des modèles de base sans effort avec Sklearn

Avant de développer un modèle de Machine Learning complexe, il est toujours judicieux de commencer par créer une base de référence.

La “baseline” sert de référence pour le modèle conçu. En outre, elle garantit que le modèle est meilleur que les prédictions aléatoires (ou fixes). Mais construire des bases de référence avec différentes stratégies (random, fixed, most frequent, etc.) peut s'avérer fastidieux.

Au lieu de cela, DummyClassifier() (et DummyRegressor()) de Sklearn rend tout cela très simple et sans effort. Tu peux sélectionner le comportement spécifique de la base de référence avec le paramètre strategy.

Pour en savoir plus, consulte la documentation.

Voici le code sur github de l’extrait de code ci-dessus.