removed unused fields for now

This commit is contained in:
Dr. Kashif Rasul
2020-03-11 13:33:51 +01:00
parent cd887ac254
commit 830a5df2aa
+5 -1
View File
@@ -11,6 +11,7 @@ from pts.transform import (
InstanceSplitter,
Transformation,
Chain,
RemoveFields,
ExpectedNumInstanceSampler,
)
from pts.dataset import FieldName
@@ -126,7 +127,10 @@ class NBEATSEstimator(PTSEstimator):
# conditioning part and a to-predict part, for each training example.
def create_transformation(self) -> Transformation:
return Chain(
[
[ RemoveFields(
field_names=[FieldName.FEAT_STATIC_REAL,
FieldName.FEAT_DYNAMIC_REAL,
FieldName.FEAT_DYNAMIC_CAT]),
InstanceSplitter(
target_field=FieldName.TARGET,
is_pad_field=FieldName.IS_PAD,