Merge pull request #48 from lewisd-benefex/patch-1

Fixed model_audit.sql to allow ARIMA_PLUS model types
kristeligt-dagblad · Sep 27, 2023 · 68045d2 · 68045d2
2 parents 2c227ca + fc40947
commit 68045d2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/macros/hooks/model_audit.sql b/macros/hooks/model_audit.sql
@@ -59,6 +59,16 @@ arima:
         - duration_ms
         - array(select as struct null as centroid_id, cast(null as float64) as cluster_radius, null as cluster_size)
     feature_info: *default_feature_info
+arima_plus:
+    training_info:
+        - training_run
+        - iteration
+        - cast(null as float64) as loss
+        - cast(null as float64) as eval_loss
+        - cast(null as float64) as learning_rate
+        - duration_ms
+        - array(select as struct null as centroid_id, cast(null as float64) as cluster_radius, null as cluster_size)
+    feature_info: *default_feature_info
 linear_reg:
     training_info: *default_training_info
     feature_info: *default_feature_info