|
@@ -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)
|
|
|
|
|
|
# 设置模型的参数
|