Add version and example

This commit is contained in:
2022-11-13 14:24:06 +01:00
parent f9f91c54a7
commit 6070c2159a
6 changed files with 52 additions and 19 deletions

View File

@@ -1,7 +1,7 @@
import sys
import time
from sklearn.model_selection import cross_val_score, StratifiedKFold
from benchmark import Discretizer
from benchmark import Datasets
from bayesclass import TAN
import warnings
@@ -15,7 +15,7 @@ start = time.time()
random_state = 17
name = sys.argv[1]
n_folds = int(sys.argv[2]) if len(sys.argv) == 3 else 5
dt = Discretizer()
dt = Datasets()
name_list = list(dt) if name == "all" else [name]
print(f"Accuracy in {n_folds} folds stratified crossvalidation")
for name in name_list: