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

FEAT_STATIC_CAT are long

Browse Source
This commit is contained in:
Kashif Rasul
2019-11-30 14:07:02 +01:00
parent e81d244832
commit dad8d09aca
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/deepar/deepar_estimator.py
+4 -2
View File
@@ -2,6 +2,8 @@ from typing import List, Optional
import numpy as np
import torch
from pts import Trainer
from pts.feature import (
TimeFeature,
@@ -88,7 +90,7 @@ class DeepAREstimator(PTSEstimator):
return Chain(
[RemoveFields(field_names=remove_field_names)] +
([SetField(output_field=FieldName.FEAT_STATIC_CAT, value=[0.0]
([SetField(output_field=FieldName.FEAT_STATIC_CAT, value=[0]
)] if not self.use_feat_static_cat else []) +
([SetField(output_field=FieldName.FEAT_STATIC_REAL, value=[0.0]
)] if not self.use_feat_static_real else []) +
@@ -96,7 +98,7 @@ class DeepAREstimator(PTSEstimator):
AsNumpyArray(
field=FieldName.FEAT_STATIC_CAT,
expected_ndim=1,
dtype=self.dtype,
dtype=np.long,
),
AsNumpyArray(
field=FieldName.FEAT_STATIC_REAL,