Skip to content

Commit 21fa740

Browse files
committed
implement get experimental data for forecasting
1 parent 5c582c8 commit 21fa740

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

tsml_eval/experiments/experiments.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -657,8 +657,15 @@ def __init__(self):
657657
pass
658658

659659
def load_experimental_data(self):
660-
# TODO IMPLEMENT
661-
return super().load_experimental_data()
660+
train = pd.read_csv(
661+
f"{self.problem_path}/{self.dataset_name}/{self.dataset_name}_TRAIN.csv", index_col=0
662+
).squeeze("columns")
663+
train = train.astype(float).to_numpy()
664+
test = pd.read_csv(
665+
f"{self.problem_path}/{self.dataset_name}/{self.dataset_name}_TEST.csv", index_col=0
666+
).squeeze("columns")
667+
test = test.astype(float).to_numpy()
668+
return train, None, test, None
662669

663670
def generate_test_preds(self, x_test, y_test):
664671
# TODO Implement this and train_preds properly

0 commit comments

Comments
 (0)