diff options
Diffstat (limited to 'bin')
| -rwxr-xr-x | bin/analyze-kconfig.py | 8 |
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: |