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

fix sample() method

Browse Source
This commit is contained in:
Kashif Rasul
2020-01-12 13:45:30 +01:00
parent 5b5a71a358
commit 7ea2331700
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/forecast.py
+1 -1
View File
@@ -530,7 +530,7 @@ class DistributionForecast(Forecast):
def to_sample_forecast(self, num_samples: int = 200) -> SampleForecast:
return SampleForecast(
samples=self.distribution.sample(num_samples),
samples=self.distribution.sample((num_samples,)),
start_date=self.start_date,
freq=self.freq,
item_id=self.item_id,