From 121ecadfcba00a9a1e8e458dd3a25b7150b882ca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alejandro=20Moreo=20Fern=C3=A1ndez?=
 <alejandro.moreo@isti.cnr.it>
Date: Wed, 18 Nov 2020 18:12:03 +0100
Subject: [PATCH] Update author_identification.py

---
 src/author_identification.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/author_identification.py b/src/author_identification.py
index 4369dc6..6c984cb 100755
--- a/src/author_identification.py
+++ b/src/author_identification.py
@@ -4,6 +4,7 @@ from data.dante_loader import load_latin_corpus, list_authors
 from data.features import *
 from model import AuthorshipVerificator
 from util.evaluation import f1_from_counters
+from sklearn.calibration import CalibratedClassifierCV
 import argparse
 
 AUTHORS_CORPUS_I = ['Dante', 'ClaraAssisiensis', 'GiovanniBoccaccio', 'GuidoFaba', 'PierDellaVigna']
@@ -63,8 +64,6 @@ def main():
             params = None
             C = args.C
 
-        from sklearn.calibration import CalibratedClassifierCV
-
         av = AuthorshipVerificator(C=C, params=params)
         av.fit(Xtr, ytr)