#!/usr/bin/env python # -*- coding: utf-8 -*- ''' Train the disambiguator ''' from quantulum3.classifier import train_classifier train_classifier()
"type": bool,
"default": True,
},
{
"dest": "lang",
"help": "language in which to train the classifier, default 'en_US'",
"type": str,
"default": "en_US",
},
]
if __name__ == "__main__":
logging.basicConfig(level=logging.INFO)
parser = argparse.ArgumentParser(
"train",
description="Train unit disambiguator based on data in quantulum "
"project folder",
)
for arg in arguments:
parser.add_argument("--{}".format(arg["dest"]), **arg)
args = parser.parse_args()
_LOGGER.info(
"Start training for language {}, {}storing the classifier".format(
args.lang, "" if args.store else "not "))
start = time.process_time()
train_classifier(store=args.store, lang=args.lang)
end = time.process_time()
_LOGGER.info("Done in {} s".format(end - start))