From 50ca593a1c29fd1e000fa11110f8768f7af1b854 Mon Sep 17 00:00:00 2001 From: Siarhei Siniak Date: Thu, 15 Jul 2021 09:59:41 +0300 Subject: [PATCH] [~] Refactor --- python/tasks/jigsaw_toxic.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/tasks/jigsaw_toxic.py b/python/tasks/jigsaw_toxic.py index 2a80df3..388839b 100644 --- a/python/tasks/jigsaw_toxic.py +++ b/python/tasks/jigsaw_toxic.py @@ -242,13 +242,14 @@ def kernel_3( # %% [code] def roc_auc(predictions,target): + import sklearn.metrics ''' This methods returns the AUC Score when given the Predictions and Labels ''' - fpr, tpr, thresholds = metrics.roc_curve(target, predictions) - roc_auc = metrics.auc(fpr, tpr) + fpr, tpr, thresholds = sklearn.metrics.roc_curve(target, predictions) + roc_auc = sklearn.metrics.auc(fpr, tpr) return roc_auc # %% [code]