Use static categorical feature in benchmark_m4 (awslabs#410)

FadhelA · Nov 29, 2019 · de4e485 · de4e485
1 parent bcc2568
commit de4e485
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/examples/benchmark_m4.py b/examples/benchmark_m4.py
@@ -74,6 +74,11 @@ def evaluate(dataset_name, estimator):
  estimator = estimator(
  prediction_length=dataset.metadata.prediction_length,
  freq=dataset.metadata.freq,
+ use_feat_static_cat=True,
+ cardinality=[
+ feat_static_cat.cardinality
+ for feat_static_cat in dataset.metadata.feat_static_cat
+ ],
  )
 
  print(f"evaluating {estimator} on {dataset}")