mirror of
https://github.com/Doctorado-ML/Stree_datasets.git
synced 2025-08-15 15:36:01 +00:00
Add sqlite config
Add sqlite to report_score
This commit is contained in:
@@ -126,11 +126,13 @@ def process_dataset(dataset, verbose, model, params):
|
||||
record = dbh.find_best(dataset, model, "gridsearch")
|
||||
hyperparameters = json.loads(record[8] if record[8] != "" else "{}")
|
||||
hyperparameters.pop("random_state", None)
|
||||
print("*" * 100)
|
||||
for random_state in random_seeds:
|
||||
random.seed(random_state)
|
||||
np.random.seed(random_state)
|
||||
kfold = KFold(shuffle=True, random_state=random_state, n_splits=5)
|
||||
clf = get_classifier(model, random_state, hyperparameters)
|
||||
print(hyperparameters)
|
||||
res = cross_validate(clf, X, y, cv=kfold, return_estimator=True)
|
||||
scores.append(res["test_score"])
|
||||
times.append(res["fit_time"])
|
||||
@@ -361,7 +363,7 @@ standardize = False
|
||||
excel,
|
||||
discretize,
|
||||
) = parse_arguments()
|
||||
# parameters = '{"splitter":"cfs","max_features":"auto"}'
|
||||
# parameters = '{"kernel":"rbf","max_features":"auto"}'
|
||||
dbh = MySQL()
|
||||
if sql:
|
||||
sql_output = open(f"{model}.sql", "w")
|
||||
|
Reference in New Issue
Block a user