def __init__(self, document, cfg_file=None, verbosity=3):
self.document = document
self.cfg = Bilbo._auto_config or cfg_file
self.config = self._get_config_parser(self.cfg)
self.pipeline = self.get_pipeline(self.config)
#TODO make a logger by default
logger = get_logger(verbosity=verbosity)
"""SVM module"""
from bilbo.libs.opts import Parser
import bilbo.components.svm.svm as s
from bilbo.utils.bilbo_logger import get_logger
if __name__ == "__main__":
args = Parser.parse_arguments()
verbosity = args.verbose
logger = get_logger(name=__name__, verbosity=args.verbose)
cfg_svm = args.cfg
data_svm = args.input
svm = s.Svm(cfg_svm)
if args.train:
svm.train(data_svm)
elif args.tag:
res = svm.predict(data_svm)
if args.verbose:
print(res)
elif args.evaluate:
res = svm.evaluate(data_svm)
else:
print("Error: Mauvais mode: choisir l'option -T / -t [train / test]")
help='Path to input file to tag')
parser.add_argument('--tagname', '-t', type=str, default='bibl',\
choices=['bibl', 'note'])
parser.add_argument('--verbose',
'-v',
action='count',
default=0,
help='Set level of verbosity')
parser.add_argument('--format', '-f', type=str,\
choices=['TEI', 'JATS', 'RESEARCH'], default="TEI", help='XML format of output file (default: %(default)s)')
args = Parser.parse_arguments()
if args.listcomponents:
mode_infos = bilbo.components.load_components()
for mname in mode_infos:
print('\t- %s : "%s"' % (mname, mode_infos[mname]['pkg'].__doc__))
exit(0)
verbosity = args.verbose
logger = get_logger(verbosity=args.verbose)
importer = Importer(args.input)
tag_separator = args.tagname
output = args.output
format_ = args.format
doc = importer.parse_xml(tag_separator)
logger.debug('Instantiate Bilbo')
bilbo = Bilbo(doc, args.cfgfile)
bilbo.run_pipeline(args.action, output, format_)