Прогнозирование видов ирисов с помощью регрессии logisit c

Question

У меня есть данный код, который сообщает вам, является ли ирис из набора данных iris vireginica или нет, в зависимости от длины лепестка и ширины лепестка. Но как я могу сделать предсказание с помощью совершенно нового цветка? 1008 * ширина чашелистика: 2,5

длина лепестка: 5,3 ширина лепестка: 2,4

Когда я пытаюсь выполнить следующий прогноз, я получаю сообщение об ошибке: ValueError: X имеет 1 функцию на образец; ожидание 2

log_reg.predict([[5.3], [2.4]])

Итак, мой вопрос: как мне сделать здесь предсказания относительно нового цветка и его вида?

Answer 1

В документации для метода predict указано, что входной аргумент должен иметь форму (n_samples, n_features), т.е. 1x2 здесь, тогда как вход в вашем случае - 2x1. Попробуйте это:

log_reg.predict([[5.3, 2.4]])