switch to sum of squared residuals as loss function

This is in line with DECART by Guo et al., 2017

1 files changed, 4 insertions, 4 deletions

diff --git a/bin/analyze-kconfig.py b/bin/analyze-kconfig.py
index 87c05f7..f7ae448 100755
--- a/bin/analyze-kconfig.py
+++ b/bin/analyze-kconfig.py
@@ -37,9 +37,9 @@ def main():
         "--attribute", choices=["rom", "ram"], default="rom", help="Model attribute"
     )
     parser.add_argument(
-        "--max-stddev",
+        "--max-loss",
         type=float,
-        help="Maximum acceptable model standard deviation for DecisionTree Leaves",
+        help="Maximum acceptable model loss for DecisionTree Leaves",
         default=10,
     )
     parser.add_argument(
@@ -65,8 +65,8 @@ def main():
     if os.path.isdir(args.model):
         data = KConfigAttributes(args.kconfig_path, args.model)
         model = KConfigModel.from_benchmark(data, args.attribute)
-        if args.max_stddev:
-            model.max_stddev = args.max_stddev
+        if args.max_loss:
+            model.max_loss = args.max_loss
         model.build_tree()
 
     else: