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

前端 未结 8 666
慢半拍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 10:51

    With neuraxle, you can easily solve this :

    p = Pipeline([
       # expected outputs shape: (n, 1)
       OutputTransformerWrapper(NumpyRavel()), 
       # expected outputs shape: (n, )
       RandomForestRegressor(**RF_tuned_parameters)
    ])
    
    p, outputs = p.fit_transform(data_inputs, expected_outputs)
    

    Neuraxle is a sklearn-like framework for hyperparameter tuning and AutoML in deep learning projects !

提交回复
热议问题