#python
#python
Вопрос:
Я пытаюсь получить вероятность в модели, но я хочу, чтобы результаты были распечатаны только как вероятность второго класса. как мне это сделать?
prediction = model.predict_proba(input)
prediction = np.array(prediction)
prediction = prediction[-1]
Результат показывает [0.66 0.33]
Но я хочу только 0.33.
Есть мысли?
Ответ №1:
Тогда последняя строка должна быть prediction = prediction[-1][-1]
или prediction = prediction[-1][1]
.
Как @Tomerikoo заставил меня заметить, вам нужен второй элемент ( ...[1]
), который также соответствует последнему ...[-1]