Ensemble Methods in Python
Román de las Heras
Data Scientist, Appodeal
base_estimatorn_estimatorsoob_scoreest_bag.oob_score_max_samples: número de muestras para cada estimador.max_features: número de variables para cada estimador.bootstrap: si se extraen muestras con reemplazo.Clasificación
from sklearn.ensemble import RandomForestClassifier
clf_rf = RandomForestClassifier(
# parameters...
)
Regresión
from sklearn.ensemble import RandomForestRegressor
reg_rf = RandomForestRegressor(
# parameters...
)
Parámetros de bagging:
n_estimatorsmax_featuresoob_scoreParámetros específicos del árbol:
max_depthmin_samples_splitmin_samples_leafclass_weight ("balanced")
Ensemble Methods in Python