Upload utils.py with huggingface_hub
Browse files
utils.py
CHANGED
|
@@ -8,7 +8,6 @@ matplotlib.use('Agg')
|
|
| 8 |
import matplotlib.pyplot as plt
|
| 9 |
import numpy as np
|
| 10 |
from sklearn.metrics import confusion_matrix, classification_report
|
| 11 |
-
import seaborn as sns
|
| 12 |
|
| 13 |
import config
|
| 14 |
|
|
@@ -96,6 +95,7 @@ def plot_confusion_matrix(y_true, y_pred, save_path):
|
|
| 96 |
y_pred: Predicted labels
|
| 97 |
save_path: Path to save the plot
|
| 98 |
"""
|
|
|
|
| 99 |
cm = confusion_matrix(y_true, y_pred)
|
| 100 |
|
| 101 |
plt.figure(figsize=(12, 10))
|
|
|
|
| 8 |
import matplotlib.pyplot as plt
|
| 9 |
import numpy as np
|
| 10 |
from sklearn.metrics import confusion_matrix, classification_report
|
|
|
|
| 11 |
|
| 12 |
import config
|
| 13 |
|
|
|
|
| 95 |
y_pred: Predicted labels
|
| 96 |
save_path: Path to save the plot
|
| 97 |
"""
|
| 98 |
+
import seaborn as sns
|
| 99 |
cm = confusion_matrix(y_true, y_pred)
|
| 100 |
|
| 101 |
plt.figure(figsize=(12, 10))
|