mirror of
https://github.com/Doctorado-ML/Stree_datasets.git
synced 2025-08-15 07:26:02 +00:00
51 lines
1.1 KiB
Python
Executable File
51 lines
1.1 KiB
Python
Executable File
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,
|
|
kernel="any",
|
|
)
|
|
for dataset in datasets:
|
|
print(f"-Cross validation on {dataset[0]}")
|
|
experiment.cross_validation(dataset[0])
|