Hice algunas actualizaciones porque las funciones en la llave “criterion” están descontinuadas, así que el código queda de esta manera:
import pandas as pd
from sklearn.model_selection import RandomizedSearchCV
from sklearn.ensemble import RandomForestRegressor
if __name__ == "__main__":
dataset = pd.read_csv('./data/felicidad.csv')
print(dataset)
X = dataset.drop(['country', 'rank', 'score'], axis=1)
y = dataset[['score']].squeeze()
reg = RandomForestRegressor()
parametros = {
'n_estimators' : range(4,16),
'criterion' : ['squared_error', 'absolute_error'],
'max_depth' : range(2,11)
}
rand_est = RandomizedSearchCV(reg, parametros , n_iter=10, cv=3, scoring='neg_mean_absolute_error').fit(X,y)
print(rand_est.best_estimator_)
print(rand_est.best_params_)
print(rand_est.predict(X.loc[[0]]))
¿Quieres ver más aportes, preguntas y respuestas de la comunidad?