fix: update for tree-based models

Author: jlwalke2

src/sasctl/utils/astore.py

@@ -383,18 +383,14 @@ def regression_target(r):
         properties["function"] = "classification"
         properties["targetVariable"] = classification_target(result)
 
-    elif algorithm == "forest":
-        properties["algorithm"] = "Random forest"
+    elif algorithm in ("forest", "gradboost", "tree-based models"):
 
-        if is_classification(result):
-            properties["function"] = "classification"
-            properties["targetVariable"] = classification_target(result)
+        if algorithm == "forest":
+            properties["algorithm"] = "Random forest"
+        elif algorithm == "gradboost":
+            properties["algorithm"] = "Gradient boosting"
         else:
-            properties["function"] = "prediction"
-            properties["targetVariable"] = regression_target(result)
-
-    elif algorithm == "gradboost":
-        properties["algorithm"] = "Gradient boosting"
+            properties["algorithm"] = "Tree-based model"
 
         if is_classification(result):
             properties["function"] = "classification"
@@ -427,7 +423,7 @@ def regression_target(r):
 
     else:
         properties["tool"] = ""
-
+    # todo: warn
     return properties