CommencerCommencer gratuitement

Entraîner un régresseur RF

Dans les exercices suivants, vous allez prédire la demande de location de vélos dans le cadre du programme Capital Bikeshare à Washington, D.C., en utilisant les données météorologiques historiques issues de l'ensemble de données Demande de vélos en libre-service disponible sur Kaggle. À cette fin, vous utiliserez l'algorithme des forêts aléatoires. Dans un premier temps, vous définirez un régresseur de fôrets aléatoires et l'ajusterez à l'ensemble d'apprentissage.

L'ensemble de données est traité pour vous et divisé en 80 % d'entraînement et 20 % de test. La matrice des fonctionnalités X_train et le tableau y_train sont disponibles dans votre espace de travail.

Cet exercice fait partie du cours

Machine learning avec des modèles arborescents en Python

Afficher le cours

Instructions

  • Importez RandomForestRegressor à partir de sklearn.ensemble.

  • Veuillez créer une instance de RandomForestRegressor appelée rf composée de 25 arbres.

  • Ajustez rf à l'ensemble d'apprentissage.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Import RandomForestRegressor
____

# Instantiate rf
rf = ____(n_estimators=____,
            random_state=2)
            
# Fit rf to the training set    
____.____(____, ____) 
Modifier et exécuter le code