Блог

Главная - Вопросы по программированию - Калибровка вероятностей нейронной сети

Калибровка вероятностей нейронной сети

#machine-learning #keras #neural-network #probability #calibration

#машинное обучение #keras #нейронная сеть #вероятность #калибровка

Вопрос:

Я хотел бы откалибровать вероятности моей модели классификации CNN. У меня есть настраиваемая перекрестная проверка, и при каждом запуске перекрестной проверки я делаю следующее:

 from sklearn.calibration import CalibratedClassifierCV
my_model.fit(X_train, y_train)
my_cccv = CalibratedClassifierCV(my_model, cv='prefit')
my_cccv.fit(X_train, y_train)
my_cccv.predict_proba(X_test)

Если я делаю это таким образом, это то же самое, что и следующее?

 from sklearn.isotonic import IsotonicRegression
my_model.fit(X_train, y_train)
ir = IsotonicRegression()
ir.fit(my_model.predict_proba(X_train),y_true)
ir.predict(X_test)

Комментарии:

1. Это не то же самое, если вы не установили method=‘isotonic’ for CalibratedClassifierCV , проверьте документы

2. @Mr.ForExample Но если я установлю метод = ‘isotonic’, тогда это будет то же самое?