moreo
/
dante-verification
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/author_identification.py
View File

@ -4,6 +4,7 @@ from data.dante_loader import load_latin_corpus, list_authors
from data.features import * from data.features import *
from model import AuthorshipVerificator from model import AuthorshipVerificator
from util.evaluation import f1_from_counters from util.evaluation import f1_from_counters
from sklearn.calibration import CalibratedClassifierCV
import argparse import argparse
AUTHORS_CORPUS_I = ['Dante', 'ClaraAssisiensis', 'GiovanniBoccaccio', 'GuidoFaba', 'PierDellaVigna'] AUTHORS_CORPUS_I = ['Dante', 'ClaraAssisiensis', 'GiovanniBoccaccio', 'GuidoFaba', 'PierDellaVigna']
@ -63,8 +64,6 @@ def main():
params = None params = None
C = args.C C = args.C
from sklearn.calibration import CalibratedClassifierCV
av = AuthorshipVerificator(C=C, params=params) av = AuthorshipVerificator(C=C, params=params)
av.fit(Xtr, ytr) av.fit(Xtr, ytr)