2015-05-26 3 views
1

Я обучил модель, используя PyNeural.Сохранение модели PyNeural на диск

Как сохранить эту обученную модель? Я пробовал pickle и sklearn.externals.joblib, но не работал. Как позже, TypeError: can't pickle NeuralNet objects

+0

Что вы пробовали? 'joblib.dump (clf, 'clfname.pkl')' это должно работать идеально. – Zero

+0

Дополнительная информация, кроме * «не работает» *? Пожалуйста, прочитайте http://stackoverflow.com/help/mcve – jonrsharpe

+0

В нем говорится: 'TypeError: не удается рассорить объекты NeuralNet' – VeilEclipse

ответ

1

Автор PyNeural здесь. Проще всего позвонить по телефону

NeuralNet.get_params() 

метод на вашем объекте NeuralNet. Это вернет кортеж, состоящий из параметров модели. Вы должны уметь рассолить и распутать этот кортеж. Для загрузки параметров в новую модель, создавать новый объект NeuralNet с теми же размерами и количеством слоев, а затем вызвать

NeuralNet.set_params(params) 

где «» PARAMS является кортежем вы сохранили.

Надеюсь, вам понравится использование PyNeural!

Смежные вопросы