From ce37ff0f57cd2f5c8a5b6d7f5ad68eaa93db739c Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Wed, 8 Jan 2020 19:47:52 +0100 Subject: [PATCH] fix taxi dataset cardinality --- pts/dataset/repository/_gp_copula_2019.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pts/dataset/repository/_gp_copula_2019.py b/pts/dataset/repository/_gp_copula_2019.py index 3e5d822..a03b055 100644 --- a/pts/dataset/repository/_gp_copula_2019.py +++ b/pts/dataset/repository/_gp_copula_2019.py @@ -96,7 +96,7 @@ datasets_info = { "taxi_30min": GPCopulaDataset( name="taxi_30min", url=root + "taxi_30min.tar.gz", - num_series=130, + num_series=1214, prediction_length=24, freq="30min", rolling_evaluations=57, @@ -111,7 +111,7 @@ def generate_gp_copula_dataset(dataset_path: Path, dataset_name: str): download_dataset(dataset_path.parent, ds_info) save_metadata(dataset_path, ds_info) - save_dataset(dataset_path / "train", ds_info) + save_dataset(dataset_path / "train" , ds_info) save_dataset(dataset_path / "test", ds_info) clean_up_dataset(dataset_path, ds_info) @@ -137,7 +137,7 @@ def save_metadata(dataset_path: Path, ds_info: GPCopulaDataset): def save_dataset(dataset_path: Path, ds_info: GPCopulaDataset): - dataset = list(FileDataset(dataset_path, freq=ds_info.freq)) + dataset = list(FileDataset(dataset_path / "*.json", freq=ds_info.freq)) shutil.rmtree(dataset_path) train_file = dataset_path / "data.json" save_to_file(