1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Machine Learning avec des modèles à base d'arbres en Python
Connected

cvičení

Entraînez votre premier arbre de régression

Dans cet exercice, vous allez entraîner un arbre de régression pour prédire la consommation mpg (miles per gallon) des voitures dans l'ensemble de données auto-mpg en utilisant les six caractéristiques disponibles.

L'ensemble de données a été préparé pour vous et divisé en 80 % pour l'entraînement et 20 % pour le test. La matrice de caractéristiques X_train et le tableau y_train sont disponibles dans votre espace de travail.

Pokyny

100 XP
  • Importez DecisionTreeRegressor depuis sklearn.tree.
  • Instanciez un DecisionTreeRegressor dt avec une profondeur maximale de 8 et min_samples_leaf réglé à 0.13.
  • Ajustez dt à l'ensemble d'entraînement.