wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-09 22:07:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix taxi dataset cardinality

Browse Source
This commit is contained in:
Kashif Rasul
2020-01-08 19:47:52 +01:00
parent 7aa6f5ff84
commit ce37ff0f57
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/dataset/repository/_gp_copula_2019.py
+3 -3
View File
@@ -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(