Python RandomForest - неизвестная ошибка метки

У меня проблемы с использованием функции подгонки RandomForest

Это мой тренировочный набор

         P1      Tp1           IrrPOA     Gz          Drz2
0        0.0     7.7           0.0       -1.4        -0.3
1        0.0     7.7           0.0       -1.4        -0.3
2        ...     ...           ...        ...         ...
3        49.4    7.5           0.0       -1.4        -0.3
4        47.4    7.5           0.0       -1.4        -0.3
... (10k rows)

Я хочу предсказать P1 благодаря всем другим переменным, используя sklearn.ensemble RandomForest

colsRes = ['P1']
X_train = train.drop(colsRes, axis = 1)
Y_train = pd.DataFrame(train[colsRes])
rf = RandomForestClassifier(n_estimators=100)
rf.fit(X_train, Y_train)

Вот ошибка, которую я получаю:

ValueError: Unknown label type: array([[  0. ],
       [  0. ],
       [  0. ],
       ..., 
       [ 49.4],
       [ 47.4],

Я не нашел ничего об этой ошибке метки, я использую Python 3.5. Любой совет будет отличной помощью!

Ответы на вопрос(3)

Ваш ответ на вопрос