Не удается загрузить модели ML на другие ПК - PullRequest
Новая
       21

Не удается загрузить модели ML на другие ПК

0 голосов
/ 23 апреля 2020

Я сохранил 4 модели машинного обучения, используя предложение здесь , сериализацию модели с засолкой:

Сохранение моделей 

pickle.dump(xgb_model, open("./classifier/classifier_fitted/XGBoost.dat", "wb")) #sklearn
pickle.dump(rf, open("./classifier/classifier_fitted/RandomForest.dat", "wb")) #sklearn
pickle.dump(ADABoost_model, open("./classifier/classifier_fitted/ADABoost.dat", "wb")) #sklearn
MLP_model.save("./classifier/classifier_fitted/MLP_model.h5") #keras

Но коллега пытается загрузить их следующим образом:

Загрузить модели 

XGBoost = pickle.load(open("./classifier/classifier_fitted/XGBoost.dat","rb"))
ADABoost = pickle.load(open("./classifier/classifier_fitted/ADABoost.dat", "rb"))
RForest = pickle.load(open("./classifier/classifier_fitted/RandomForest.dat", "rb"))
MLP = load_model("./classifier/classifier_fitted/MLP_model.h5")

Но он получает несколько ошибок, чтобы загрузить их, другой сотрудник пытается сделать то же самое, у него те же ошибки. У меня хорошо работает, были ли у кого-нибудь подобные проблемы?

...