Add nodes, leaves, depth to mysql

Add nodes, leaves, depth, samples, features and classes to analysis
This commit is contained in:
2021-03-24 12:38:11 +01:00
parent ae6e2ba374
commit 66bceff179
5 changed files with 124 additions and 74 deletions

View File

@@ -176,6 +176,7 @@ class BD(ABC):
accuracy,
time_spent,
parameters,
complexity,
) -> None:
"""Create a record in MySQL database
@@ -187,8 +188,8 @@ class BD(ABC):
command_insert = (
"replace into results (date, time, type, accuracy, "
"dataset, classifier, norm, stand, parameters, accuracy_std, "
"time_spent, time_spent_std) values (%s, %s, "
"%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)"
"time_spent, time_spent_std, nodes, leaves, depth) values (%s, %s,"
" %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)"
)
now = datetime.now()
date = now.strftime("%Y-%m-%d")
@@ -206,6 +207,9 @@ class BD(ABC):
accuracy[1],
time_spent[0],
time_spent[1],
complexity["nodes"],
complexity["leaves"],
complexity["depth"],
)
cursor = database.cursor()
cursor.execute(command_insert, values)
@@ -319,7 +323,9 @@ class Outcomes(BD):
self._table = "outcomes"
super().__init__(host=host, model=model)
def store(self, dataset, normalize, standardize, parameters, results):
def store(
self, dataset, normalize, standardize, parameters, results, complexity
):
outcomes = ["fit_time", "score_time", "train_score", "test_score"]
data = ""
for index in outcomes:
@@ -350,6 +356,7 @@ class Outcomes(BD):
float(np.std(results["fit_time"])),
],
parameters,
complexity,
)
def report(self, dataset, exclude_params):