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()