You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
770 B
24 lines
770 B
#!/usr/bin/env python3 |
|
import argparse |
|
import json |
|
|
|
parser = argparse.ArgumentParser(description="General Purpose Text Classifier") |
|
parser.add_argument('model', help='model to use') |
|
parser.add_argument('-c', '--compile', help='compile raw model model to outfile', metavar='outfile') |
|
args = parser.parse_args() |
|
|
|
import gptc # PEP 8 violation, but don't fix it |
|
# Way better for performance of argparse checking |
|
|
|
with open(args.model, 'r') as f: |
|
raw_model = json.load(f) |
|
if args.compile: |
|
with open(args.compile, 'w+') as f: |
|
json.dump(gptc.compile(raw_model), f) |
|
else: |
|
classifier = gptc.Classifier(raw_model) |
|
if sys.stdin.isatty(): |
|
text = input('Text to analyse: ') |
|
else: |
|
text = sys.stdin.read() |
|
print(classifier.check(text))
|
|
|