A column-vector y was passed when a 1d array was expected

前端 未结 8 659
慢半拍i
慢半拍i 2020-12-12 10:40

I need to fit RandomForestRegressor from sklearn.ensemble.

forest = ensemble.RandomForestRegressor(**RF_tuned_parameters)
model = fo         


        
8条回答
  •  挽巷
    挽巷 (楼主)
    2020-12-12 11:06

    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)

    .ravel will convert that array shape to (n, )

提交回复
热议问题