Ошибка значения: неверное количество элементов прошло 2, размещение подразумевает 1

Этот код отлично работает ...

forest1 = RandomForestClassifier()
forest1.fit(train[['Random Forest Score','lr','neural']],train['target'])

Но когда я пытаюсь предсказать,

test['target'] = forest1.predict_proba(test[['Random Forest Score','lr','neural']])

Показывает ошибку ..

ValueError: неверное количество элементов прошло 2, размещение подразумевает 1


pandas machine-learning valueerror
person Amogh Badugu    schedule 03.10.2017    source источник

Ответы (1)


arrow_upward
3
arrow_downward

forest1.predict_proba (.. .) предсказывает класс вероятности для X.

Он возвращает массив shape = [n_samples, n_classes] или список n_outputs таких массивов, если n_outputs> 1. Вероятности классов входных выборок. Порядок классов соответствует таковому в атрибуте classes_.

test['target'] ожидает вектор (одномерный массив)

Попробуйте использовать _3 _ вместо predict_proba:

test['target'] = forest1.predict(test[['Random Forest', 'Score','lr','neural']])
person MaxU    schedule 03.10.2017