From bfdc475372bb61161aeb3d2caa54549d0d5a3af7 Mon Sep 17 00:00:00 2001
From: driskerr <7308323+driskerr@users.noreply.github.com>
Date: Mon, 17 Jun 2024 14:47:56 -0700
Subject: [PATCH] set up model_audit to handle PCA use models

---
 macros/hooks/model_audit.sql | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/macros/hooks/model_audit.sql b/macros/hooks/model_audit.sql
index c2ea990..80ba8e1 100644
--- a/macros/hooks/model_audit.sql
+++ b/macros/hooks/model_audit.sql
@@ -92,7 +92,16 @@ logistic_reg:
     feature_info: *default_feature_info
     weights: ['*']
 matrix_factorization: *default
-pca: *default
+pca:
+    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
 autoencoder: *default
 tensorflow: {}
 tensorflow_lite: {}