added roc auc metrics

seenstevo · seenstevo · commit 48132b9015b5 · 2023-02-23T08:27:18.000+01:00
diff --git a/toolkit/plot.py b/toolkit/plot.py
@@ -9,6 +9,7 @@
 import plotly.offline as py
 from wordcloud import STOPWORDS
 import plotly.express as px
+from sklearn.metrics import auc, roc_curve
 
 
 def plot_multiclass_prediction_image(df, row_index: int, X_test: Union[pd.DataFrame, np.ndarray], prediction_col: str = 'Top Prediction', label_col: str = 'Label'):
