炼丹笔记:记录我们的成长轨迹
LightGBM如何保存模型?
用lgb保存模型遇到了几个坑,在这里记录一下。
在用Lightgbm.LGBMClassifier训练生成模型时,Scikit-learn 官网上建议的两种方式:
1.pickle方式
这里我写了保存和加载两种方式:
import pickle
def pkl_save(filename,file):
output = open(filename, 'wb')
pickle.dump(file, output)
output.close()
def pkl_load(filename):
pkl_file = open(filename, 'rb')
file = pickle.load(pkl_file)
pkl_file.close()
return file
2.joblib方式
from sklearn.externals import joblib
joblib.dump(model_lgb, 'dota_model.pkl')
clf = joblib.load('dota_model.pkl')