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

typo

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-03-09 10:53:45 +01:00
parent 37fdb0e5b7
commit 539108a331
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/dataset/artificial.py
+2 -2
View File
@@ -72,7 +72,7 @@ class ArtificialDataset:
return TrainDatasets(
metadata=self.metadata,
train=ListDataset(self.train, self.freq),
test=ListDataset(self.test, self.freq, shuffle=False),
test=ListDataset(self.test, self.freq),
)
@@ -697,7 +697,7 @@ class RecipeDataset(ArtificialDataset):
return TrainDatasets(
metadata=metadata,
train=ListDataset(train_data, metadata.freq),
test=ListDataset(test_data, metadata.freq, is_train=False),
test=ListDataset(test_data, metadata.freq),
)
pts/dataset/multivariate_grouper.py
+1 -1
View File
@@ -144,7 +144,7 @@ class MultivariateGrouper:
all_entries.append(grouped_data)
return ListDataset(
all_entries, freq=self.frequency, one_dim_target=False, is_train=False
all_entries, freq=self.frequency, one_dim_target=False
)
def _align_data_entry(self, data: DataEntry) -> np.array: