Describe the bug
The same classes passed through sklearn label encoder are shown inverted in classification report. I tried the same with confusion matrix and classes are correctly shown
To Reproduce
# CLASSIFICATION REPORT
visualizer = ClassificationReport(
pipeline,
classes = label_encoder.classes_,
cmap="YlGn",
support=True
)
visualizer.fit(X_train, y_train)
visualizer.score(X_test, y_test)
# CONFUSION MATRIX
cm = ConfusionMatrix(
pipeline,
classes= label_encoder.classes_,
cmap='Greens'
)
cm.fit(X_train, y_train)
cm.score(X_test, y_test)
Dataset
Whatever dataset you prefer
Expected behavior
Coherent classes visualization
Desktop (please complete the following information):
- OS: Cloud Jupiter notebook
- Python Version 3.11
- Yellowbrick Version 1.5
Describe the bug
The same classes passed through sklearn label encoder are shown inverted in classification report. I tried the same with confusion matrix and classes are correctly shown
To Reproduce
Dataset
Whatever dataset you prefer
Expected behavior
Coherent classes visualization
Desktop (please complete the following information):