如何加载通过save_binary保存的LightGBM Dataset数据集

你可以直接通过lgb.Dataset()构造方法传入二进制文件路径完成加载，不需要额外调用其他接口。

你提供的保存代码存在小笔误，import numpy as 后缺少别名np，修正后的完整保存+加载示例如下：

import numpy as np
import lightgbm as lgb

# 构造并保存数据集
data = lgb.Dataset(np.array([[1,2],[12,2]]))
data.save_binary('test.bin')

# 加载已保存的二进制数据集
loaded_data = lgb.Dataset('test.bin')

补充说明 #

• 加载得到的数据集和原始数据集完全等价，可以直接传入lgb.train()等训练接口使用
• 若原始数据集绑定了标签、样本权重、分组信息等元数据，保存时会同步写入二进制文件，加载后会自动恢复，无需额外传入参数

内容的提问来源于stack exchange，提问作者DvdG