Я предполагаю, что у вас уже есть данные и лабораторные работы, и вы разделили их на обучающие и тестовые данные и метки, и вы получили список прогнозирования, используя следующий метод. Затем вам нужно получить вероятности из модели, позвонив по номеру clf.predict_proba(data_test)
, как показано ниже.
import numpy as np
from sklearn.metrics import log_loss
from sklearn.linear_model import LogisticRegression
#test_list = [1,0,0,1,0,.....]
#prediction_list = [1,1,0,1,0......]
# Model learning and prediction
clf = LogisticRegression()
prediction_list = clf.predict(data_test)
pred_probabilities = clf.predict_proba(data_test)
# Evaluation of the prediction
print("The binary cross entropy loss is : %f" % log_loss(labels_test, pred_probabilities))
Я все еще новичок в машинном обучении, поэтому возьмите это с крошкой соли.