mirror of
https://github.com/Doctorado-ML/Stree_datasets.git
synced 2025-08-17 00:16:09 +00:00
Fix some mistakes
Add complexity to crossval Change header of tex file Add complexity fields to report_score Add commented final date in database
This commit is contained in:
@@ -182,6 +182,13 @@ class Experiment:
|
||||
),
|
||||
sort_keys=True,
|
||||
)
|
||||
if type(model).__name__ == "Stree":
|
||||
best_model = grid_search.best_estimator_
|
||||
nodes, leaves = best_model.nodes_leaves()
|
||||
depth = best_model.depth_
|
||||
else:
|
||||
nodes = leaves = depth = 0
|
||||
complexity = dict(nodes=nodes, leaves=leaves, depth=depth)
|
||||
hyperparams.store(
|
||||
dataset,
|
||||
time_spent,
|
||||
@@ -190,6 +197,7 @@ class Experiment:
|
||||
normalize,
|
||||
standardize,
|
||||
grid_type,
|
||||
complexity,
|
||||
)
|
||||
if self._num_warnings > 0:
|
||||
print(f"{self._num_warnings} warnings have happend")
|
||||
|
Reference in New Issue
Block a user