(self)
| 62 | self.assertEqual(1, gbm.best_iteration) |
| 63 | |
| 64 | def load_datasets(self): |
| 65 | df_train = pd.read_csv('/input/tests/data/lgb_train.csv', header=None, sep='\t') |
| 66 | df_test = pd.read_csv('/input/tests/data/lgb_test.csv', header=None, sep='\t') |
| 67 | |
| 68 | y_train = df_train[0] |
| 69 | y_test = df_test[0] |
| 70 | X_train = df_train.drop(0, axis=1) |
| 71 | X_test = df_test.drop(0, axis=1) |
| 72 | |
| 73 | lgb_train = lgb.Dataset(X_train, y_train) |
| 74 | lgb_eval = lgb.Dataset(X_test, y_test, reference=lgb_train) |
| 75 | |
| 76 | return (lgb_train, lgb_eval) |