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. Любой совет будет отличной помощью!