Instancier le modèle
Dans la série d'exercices suivante, vous allez diagnostiquer les problèmes de biais et de variance d'un arbre de régression. L'arbre de régression que vous allez définir dans cet exercice servira à prédire la consommation en mpg des voitures du jeu de données auto à partir de toutes les caractéristiques disponibles.
Nous avons déjà traité les données et chargé dans votre espace de travail la matrice de caractéristiques X et le tableau y. De plus, la classe DecisionTreeRegressor a été importée depuis sklearn.tree.
Cette activité fait partie du cours
Machine Learning avec des modèles à base d'arbres en Python
Instructions de l’exercice
- Importez
train_test_splitdesklearn.model_selection. - Divisez les données en 70 % entraînement et 30 % test.
- Instanciez un
DecisionTreeRegressoravec une profondeur maximale de 4 etmin_samples_leafréglé à 0.26.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Import train_test_split from sklearn.model_selection
____
# Set SEED for reproducibility
SEED = 1
# Split the data into 70% train and 30% test
X_train, X_test, y_train, y_test = ____(____, ____, test_size=____, random_state=SEED)
# Instantiate a DecisionTreeRegressor dt
dt = ____(____=____, ____=____, random_state=SEED)