罗俊辉 1 年間 前
コミット
49ba51f6db
1 ファイル変更1 行追加1 行削除
  1. 1 1
      main.py

+ 1 - 1
main.py

@@ -42,7 +42,7 @@ with open("whole_data/y_data.json") as f2:
     y_test = np.array(y_list[10000:])
 
 # 创建LightGBM数据集
-train_data = lgb.Dataset(X_train, label=y_train, categorical_features=['uid', 'type', 'channel'])
+train_data = lgb.Dataset(X_train, label=y_train, categorical_feature=['uid', 'type', 'channel'])
 test_data = lgb.Dataset(X_test, label=y_test, reference=train_data)
 
 # 设置模型的参数