LogisticRegression

Логистическая регрессия для классификации.

СигнатураLogisticRegression(C=1.0, penalty='l2', max_iter=100)

Несмотря на название, это классификатор: предсказывает вероятности классов через логистическую функцию. Параметр C обратно пропорционален силе регуляризации. Для многоклассовых задач работает автоматически.

from sklearn.linear_model import LogisticRegression

clf = LogisticRegression(max_iter=1000)
clf.fit(X_train, y_train)
print(clf.predict_proba(X_test)[:3])

← Все записи: scikit-learn