penguins-decision-tree / train_metadata.json
gusdelact's picture
DecisionTreeClassifier v0.1.0
6e8e22e verified
Raw
History Blame Contribute Delete
1.54 kB
{
"design_doc": "notes/02_design_modeling.md",
"estimator": "DecisionTreeClassifier",
"model_type": "decision_tree_classifier",
"trained_at": "2026-05-18T14:31:31.534278",
"training_seconds": 4.203144073486328,
"random_state": 42,
"features": [
"Culmen Length (mm)",
"Culmen Depth (mm)",
"Flipper Length (mm)",
"Body Mass (g)",
"Island_Biscoe",
"Island_Dream",
"Island_Torgersen",
"Sex_FEMALE",
"Sex_MALE",
"Clutch Completion_No",
"Clutch Completion_Yes"
],
"target": "Species",
"n_samples_train": 258,
"n_features": 11,
"search_strategy": "GridSearchCV",
"cv_folds": 5,
"scoring": "f1_weighted",
"param_grid": {
"max_depth": [
3,
4,
5,
6,
8,
10,
null
],
"min_samples_split": [
2,
5,
10
],
"min_samples_leaf": [
1,
2,
5
],
"criterion": [
"gini",
"entropy"
]
},
"best_params": {
"criterion": "gini",
"max_depth": 5,
"min_samples_leaf": 1,
"min_samples_split": 5
},
"params": {
"criterion": "gini",
"max_depth": 5,
"min_samples_leaf": 1,
"min_samples_split": 5
},
"best_cv_score": 0.973031605176191,
"best_cv_score_std": 0.019704745060290966,
"best_fold_scores": [
0.9808021038790269,
0.9615384615384616,
0.9807921654969955,
0.9420252949664715,
1.0
],
"best_train_score_mean": 0.9932328656107854,
"model_path": "models/model.joblib",
"tree_depth": 5,
"tree_n_leaves": 10
}