Vous trouverez un exemple de code pour apprendre simplement et rapidement un modèle avec Scikit-Learn.
Le fichier src/prepare.py charge les données et les arrange de manière à pouvoir être traitées.
Le fichier src/main.py montre comment séparer les données de départ en deux sous ensembles:
- Le
train setpour apprendre le modèle avec la fonctionfit() - Le
test setpour réaliser des prédictions et comaprer le résultat avec la réalité, via la fonctionscore()
Installer les dépendances:
pip install -U requirements.txtExécuter:
python src/main.py