def get_classification_report_for_evaluation(base_model: BaseModel) -> Dict:
y_pred, y_test = base_model.test_model()
report = classification_report(y_test,
y_pred,
target_names=_CLASS_NAMES,
output_dict=True)
return report
def evaluate_model(base_model: BaseModel) -> None:
y_pred, y_test = base_model.test_model()
print(classification_report(y_test, y_pred, target_names=_CLASS_NAMES))
cm = confusion_matrix(y_test, y_pred)
df_cm = pd.DataFrame(cm, index=_CLASS_NAMES, columns=_CLASS_NAMES)
show_confusion_matrix(df_cm)
