Commit Inicial

cross_all.py

@@ -0,0 +1,54 @@
+import argparse
+from typing import Tuple
+
+from experimentation.Sets import Datasets
+from experimentation import Experiment
+
+
+def parse_arguments() -> Tuple[str, str, str, bool, bool]:
+    ap = argparse.ArgumentParser()
+    ap.add_argument(
+        "-H",
+        "--host",
+        type=str,
+        choices=["develop", "galgo"],
+        required=False,
+        default="develop",
+    )
+    ap.add_argument(
+        "-m",
+        "--model",
+        type=str,
+        choices=["stree", "adaBoost", "bagging", "odte"],
+        required=False,
+        default="stree",
+    )
+    ap.add_argument(
+        "-S",
+        "--set-of-files",
+        type=str,
+        choices=["aaai", "tanveer"],
+        required=False,
+        default="aaai",
+    )
+    args = ap.parse_args()
+    return (
+        args.host,
+        args.model,
+        args.set_of_files,
+    )
+
+
+(
+    host,
+    model,
+    set_of_files,
+) = parse_arguments()
+datasets = Datasets(False, False, set_of_files)
+clf = None
+experiment = Experiment(
+    random_state=1, model=model, host=host, set_of_files=set_of_files
+)
+for dataset in datasets:
+    print(f"-Cross validation on {dataset[0]}")
+    experiment.cross_validation(dataset[0])