mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-07-24 13:20:07 +08:00
fix prediction length
This commit is contained in:
@@ -249,7 +249,7 @@ class SampleForecast(Forecast):
|
||||
"""
|
||||
Time length of the forecast.
|
||||
"""
|
||||
return self.samples.shape[-1]
|
||||
return self.samples.shape[1]
|
||||
|
||||
@property
|
||||
def mean(self):
|
||||
|
||||
Reference in New Issue
Block a user