Add models oc1 and cart

2025-08-15 23:46:03 +00:00 · 2021-02-28 18:46:58 +01:00
parent 091edbe3dc
commit df42c0df74
2 changed files with 57 additions and 1 deletions
--- a/experiment.py
+++ b/experiment.py
@@ -20,7 +20,7 @@ def parse_arguments() -> Tuple[str, str, str, str, str, bool, bool, dict]:
        "-m",
        "--model",
        type=str,
-        choices=["stree", "adaBoost", "bagging", "odte"],
+        choices=["stree", "adaBoost", "bagging", "odte", "oc1", "cart"],
        required=False,
        default="stree",
    )