Closed wangbingnan136 closed 5 years ago
@wangbingnan136 You can dump and load your model like the following:
where estimator_class
can be RGFRegressor, RGFClassifier, FastRGFRegressor or FastRGFClassifier.
I can't understand why keras
is in your traceback. You can post the whole code, if you wish to find the mistake, because
from sklearn.externals import joblib
joblib.dump(clf,'fastrgf_model.pkl')
clf=joblib.load('fastrgf_model.pkl')
works fine.
Thank u very much,It's my mistake hh!
No problem! Glad that you've solved the issue!
from sklearn.externals import joblib joblib.dump(clf,'fastrgf_model.pkl')
clf=joblib.load('fastrgf_model.pkl')
KeyError Traceback (most recent call last)