1. 程式人生 > >lightgbm : wrong size of feature_names

lightgbm : wrong size of feature_names

今天在訓練模型的時候,遇到這樣一個錯誤,已經訓練好模型了,但卻報錯:wrong size,以前也沒有遇到過,檢視lightgbm的原始碼,發現可能是這樣原因

 

 

檢查了特徵中是否存在“ ”空格的錯誤後,發現並不是這個問題。

請教朋友後知道,原來lgb在訓練模型的時候要用numy格式,也就是在生成lgb格式資料的時候要這樣來操作

如果沒有寫 .values 可能會報錯