def cli_classify():
parser = argparse.ArgumentParser(description=_classify_descr,
formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument('dataset', nargs=1, metavar='DATASET',
help="path to pre-trained dataset")
parser.add_argument('files', nargs='*', metavar='FILE',
help="paths to files to classify")
args = parser.parse_args()
classifier = Classifier(args.dataset[0])
if not args.files:
args.files = ['-']
for path in args.files:
if path == '-':
text = ''.join(sys.stdin.readlines())
else:
with open(path, 'r', encoding='utf-8', errors='ignore') as f:
text = ''.join(f.readlines())
print('{0}\t{1}'.format(path, classifier.classify(text)))
评论列表
文章目录