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

This commit is contained in:
nuluh
2025-08-31 13:01:04 +07:00
parent cdcd056102
commit d8482988ff

View File

@@ -2,7 +2,7 @@ import numpy as np
import pandas as pd
import os
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
def create_ready_data(
@@ -159,6 +159,7 @@ def train_and_evaluate_model(
# Continue despite export error
result["success"] = True
result["classification_report"] = classification_report(y_test, y_pred, output_dict=True)
return result
except Exception as e: