edaflow.ml.plot_confusion_matrix

edaflow.ml.plot_confusion_matrix(model: BaseEstimator, X_val: DataFrame, y_val: Series, normalize: bool = False, title: str | None = None, figsize: Tuple[int, int] = (8, 6)) → Figure

Plot confusion matrix for a classification model.

Parameters:

modelBaseEstimator

Fitted classification model

X_valpd.DataFrame

Validation features

y_valpd.Series

Validation target

normalizebool, default=False

Whether to normalize the confusion matrix

titlestr, optional

Plot title

figsizeTuple[int, int], default=(8, 6)

Figure size

Returns:

plt.Figure

The matplotlib figure