I need to fit RandomForestRegressor from sklearn.ensemble.
RandomForestRegressor
sklearn.ensemble
forest = ensemble.RandomForestRegressor(**RF_tuned_parameters) model = fo
Change this line:
model = forest.fit(train_fold, train_y)
to:
model = forest.fit(train_fold, train_y.values.ravel())
Edit:
.values will give the values in an array. (shape: (n,1)
.values
.ravel will convert that array shape to (n, )
.ravel