nuluh/thesis
1
0
Fork 0
Code Issues 47 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

feat(ml): add classification report generation to model evaluation to show all metrics during training

Browse Source
This commit is contained in:
nuluh
2025-08-31 13:01:04 +07:00
parent cdcd056102
commit d8482988ff
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
code/src/ml/model_selection.py
View File

@@ -2,7 +2,7 @@ import numpy as np
import pandas as pd import pandas as pd
import os import os
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay, classification_report
from joblib import load from joblib import load
def create_ready_data( def create_ready_data(
@@ -159,6 +159,7 @@ def train_and_evaluate_model(
# Continue despite export error # Continue despite export error
result["success"] = True result["success"] = True
result["classification_report"] = classification_report(y_test, y_pred, output_dict=True)
return result return result
except Exception as e: except Exception as e: