kingfengji / gcForest

This is the official implementation for the paper 'Deep forest: Towards an alternative to deep neural networks'
http://lamda.nju.edu.cn/code_gcForest.ashx
1.31k stars 429 forks source link

How to save and load models? #47

Open 783572361 opened 5 years ago

783572361 commented 5 years ago

How to save and load models?

zhaopanpan73 commented 5 years ago

I have the same question as you

RizhaoCai commented 5 years ago

How to save and load models?

SAME QUESTION. How to save a model and load it? Have you found the way?

RizhaoCai commented 5 years ago

How to save and load models?

class SKlearnBaseClassifier(BaseClassifierWrapper): def _load_model_from_disk(self, cache_path): return joblib.load(cache_path)

def _save_model_to_disk(self, clf, cache_path):
    joblib.dump(clf, cache_path)

you can refer to this: image