如何在lightgbm树拆分阈值中使用float(不是double)?

时间:2019-07-19 13:58:05

标签: lightgbm

Xgboost使用float来作为分割阈值,为了提高准确性,lightgbm用double代替了float。如何将其切换回浮动状态?

我将所有训练数据都转换为float32,并将其输入到lightgbm中。

x = np.array(x, dtype=np.int)
y = np.array(y, dtype=np.int)
gbm.fit(x, y)

仍然,所有拆分阈值都加倍。

0 个答案:

没有答案