Ошибка при загрузке двух разных моделей соления в Django - PullRequest
0 голосов
/ 26 февраля 2020

У меня есть Django APP, где в view.py я включил две разные модели загрузки маринада следующим образом:

 model = pickle.load(open(os.path.join(settings.STATIC_ROOT,'randomforest.sav'),"rb"))
 model2 = pickle.load(open(os.path.join(settings.STATIC_ROOT,'svm.sav'),"rb"))

, а затем я установил две модели следующим образом:

classification = model.predict(X_test)
classification2 = model2.predict(X_test2)

Но теперь, когда я печатаю результат прогноза, возвращается два классификатора:

'from model 1' [1. 0.]
'from model 2' [1. 0.]

, даже если классификатор SVM должен возвращать только классификацию как:

[2]

it Кажется, что вторая модель (то есть, model2) не загружена или не вызвана должным образом. Как я могу решить эту проблему?

Просто чтобы прояснить: используя те же логи c, например, в colab, я получаю следующие правильные результаты:

[[1. 0.]]
[2]
...