相关文章推荐

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()
 
推荐文章