Update author_identification.py

This commit is contained in:
Alejandro Moreo Fernandez 2020-11-18 18:12:03 +01:00 committed by GitHub
parent b4209a15a0
commit 121ecadfcb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions

View File

@ -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)