Я пытаюсь получить предсказанный класс, используя предсказанную вероятность, но точность резко меняется, учитывая класс [-1, 1] соответственно.
Вот код, который я использую для преобразования вероятности в предсказаннуюкласс.
y_pred_prob = []
for i in Y_prob:
if i[0] > i[1]:
y_pred_prob.append(-1)
else:
y_pred_prob.append(1)
Когда я использую предсказанный класс, точность составляет 0,8983, но когда я использую y_pred_prob
, точность изменяется до 0,6241