lightgbm : wrong size of feature_names

今天在训练模型的时候,遇到这样一个错误,已经训练好模型了,但却报错:wrong size,以前也没有遇到过,查看lightgbm的源码,发现可能是这样原因

 

检查了特征中是否存在“ ”空格的错误后,发现并不是这个问题。

请教朋友后知道,原来lgb在训练模型的时候要用numy格式,也就是在生成lgb格式数据的时候要这样来操作

如果没有写 .values 可能会报错

猜你喜欢

转载自blog.csdn.net/a1066196847/article/details/82757260