search

kalekiu / easyesn

Python library for Reservoir Computing using Echo State Networks
129 stars 37 forks source link

Model storing #10

Closed Shemka closed 5 years ago

Shemka commented 5 years ago

Hello, how can I save already trained model?

ikespand commented 5 years ago

As this library is build upon Scikit-learn, so you can dump trained model data (i.e. after esn.fit) using joblib.

from sklearn.externals import joblib
joblib.dump(esn, "esn_model.sav")

You can then load model as

loaded_model = joblib.load( "esn_model.sav")
Y_new_prediction = loaded_model.predict(X_new)