ACTIVITAT 5.4: UN PROBLEMA DE REGRESSIÓ

Els arbres de decisió no només serveixen per classificar, també els podem fer servir per problemes de regressió (problemes on volem predir un valor numèric). Per fer-ho haureu d'utilitzar AQUEST mètode i, per tant, abans l'hem d'importar. P.ex:

from sklearn.tree import DecisionTreeRegressor

model2 = DecisionTreeRegressor().fit(features_train, target_train)

Utilitzant el dataset d'iris o el dels pingüins fes un model de regressió per predir una característica numèrica. P.ex: pots predir l'amplada del pètal a partir de la seva llargada i l'amplada i llargada del sèpal. En aquest cas podem obviar l'espècie d'iris.