Keras: в чем разница между model.predict и model.predict_proba

Я обнаружил, что model.predict и model.predict_proba дают идентичную 2D-матрицу, представляющую вероятности для каждой категории для каждой строки.

В чем разница между двумя функциями?

Ответы

Ответ 1

предсказать

predict(self, x, batch_size=32, verbose=0)

Создает выходные предсказания для входных выборок, обрабатывая образцы в пакетном режиме.

Аргументы

x: the input data, as a Numpy array.
batch_size: integer.
verbose: verbosity mode, 0 or 1.

Возвращает

A Numpy array of predictions.

predict_proba

predict_proba(self, x, batch_size=32, verbose=1)

Генерирует предсказания вероятности класса для партии входных выборок по пакету.

Аргументы

x: input data, as a Numpy array or list of Numpy arrays (if the model has multiple inputs).
batch_size: integer.
verbose: verbosity mode, 0 or 1.

Возвращает

A Numpy array of probability predictions.

Изменить. В последней версии keras предсказать и прогнозировать_proba - то же самое, то есть оба дают вероятности. Чтобы получить метки классов, используйте prec_classes. Документация не обновляется. (адаптировано из комментария Avijit Dasgupta)