From dad8d09aca7653de4e7cce32bebd87cfb678ff86 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Sat, 30 Nov 2019 14:07:02 +0100 Subject: [PATCH] FEAT_STATIC_CAT are long --- pts/model/deepar/deepar_estimator.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pts/model/deepar/deepar_estimator.py b/pts/model/deepar/deepar_estimator.py index 8a26b8b..6b741a1 100644 --- a/pts/model/deepar/deepar_estimator.py +++ b/pts/model/deepar/deepar_estimator.py @@ -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,