|
@@ -14,6 +14,8 @@ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_
|
|
|
train_data = lgb.Dataset(X_train, label=y_train)
|
|
train_data = lgb.Dataset(X_train, label=y_train)
|
|
|
test_data = lgb.Dataset(X_test, label=y_test, reference=train_data)
|
|
test_data = lgb.Dataset(X_test, label=y_test, reference=train_data)
|