From ee389fc21e87a373d2d7d3ed1c4047165344bad8 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Wed, 16 Sep 2020 12:55:48 +0200 Subject: switch to sum of squared residuals as loss function This is in line with DECART by Guo et al., 2017 --- bin/analyze-kconfig.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'bin') 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: -- cgit v1.2.3