Прогнозирование модели ONNX - прогнозирование классов - PullRequest
0 голосов
/ 20 апреля 2020

У меня следующая проблема: я использую среду выполнения ONNX в C# для загрузки модели hdf5, которую я обучал с тенор-потоком кера.

Теперь я хочу делать прогнозы, что тоже работает.

Проблема в том, что я могу выбрать при использовании тензор-поток / керас для прогнозирования класса или просто для прогнозирования.

evaluateTest = model2.model.predict(X_test)
print(evaluateTest)
loaded_model.predict_classes(X_test)

Результаты разные, я думаю из-за подгонки или масштабирования данных для набор данных test и train.

В c# он ведет себя как первый прогноз и дает мне, например, 0.4300000

Это тот же результат, что и в коде первого примера. Второй пример в коде показывает мне класс: 0 или 1.

Есть ли способ достичь этого и в C#?

...