"ValueError: найден массив с dim 3. Ожидается оценщик <= 2" - PullRequest
Новая
       24

"ValueError: найден массив с dim 3. Ожидается оценщик <= 2"

0 голосов
/ 17 января 2020

Я пытаюсь запустить простую модель регрессии Logisti c, но я продолжаю получать эту ошибку "ValueError: Найдено массив с dim 3. Ожидается оценщик <= 2." Мой X_train имеет 2 столбца: возраст и предполагаемая зарплата. Как правильно получить прогноз? </p> 

from sklearn.linear_model import LogisticRegression
classifier = LogisticRegression(random_state = 0)
classifier.fit(X_train, y_train)

# Predicting
val1= input("Age: ")
val2 = input("Estimated Salary: ")
y_pred = classifier.predict([np.array([val1]).reshape(1, 1), np.array([val2]).reshape(1, 1)])
...