Назад к вопросам
JuniorТеорияЧасто
Что выведет эта программа?
import numpy as np
from sklearn.linear_model import LogisticRegression
X = np.array([[0.0], [2.0], [4.0], [6.0], [8.0], [10.0]])
y = np.array([0, 0, 0, 1, 1, 1])
model = LogisticRegression()
model.fit(X, y)
proba = model.predict_proba(np.array([[9.0]]))
print(round(proba[0, 0], 2), model.predict(np.array([[9.0]]))[0])0.02 1
- Ждать
0.98: столбцыpredict_probaидут в порядкеmodel.classes_, поэтому в столбце0лежит вероятность класса0, а класс1— в столбце1. - Ждать, что
predictвернёт вероятность: он возвращает метку, здесь1, потому что вероятность класса1больше 0.5.
- Как получить вероятность класса
1для каждой строкиX_test? - Что хранит
model.classes_и как он задаёт порядок столбцов?