Update author_identification.py
This commit is contained in:
parent
b4209a15a0
commit
121ecadfcb
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue