Format code with black
This commit is contained in:
parent
e272ab42d1
commit
a207e281e7
|
@ -33,7 +33,9 @@ print(
|
|||
)
|
||||
|
||||
|
||||
classifier = gptc.Classifier(gptc.compile(raw_model, max_ngram_length), max_ngram_length)
|
||||
classifier = gptc.Classifier(
|
||||
gptc.compile(raw_model, max_ngram_length), max_ngram_length
|
||||
)
|
||||
print(
|
||||
"Average classification time over",
|
||||
classify_iterations,
|
||||
|
|
|
@ -6,19 +6,34 @@ import json
|
|||
import sys
|
||||
import gptc
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(
|
||||
description="General Purpose Text Classifier", prog="gptc"
|
||||
)
|
||||
subparsers = parser.add_subparsers(dest="subparser_name", required=True)
|
||||
|
||||
compile_parser = subparsers.add_parser("compile", help="compile a raw model")
|
||||
compile_parser = subparsers.add_parser(
|
||||
"compile", help="compile a raw model"
|
||||
)
|
||||
compile_parser.add_argument("model", help="raw model to compile")
|
||||
compile_parser.add_argument("--max-ngram-length", "-n", help="maximum ngram length", type=int, default=1)
|
||||
compile_parser.add_argument(
|
||||
"--max-ngram-length",
|
||||
"-n",
|
||||
help="maximum ngram length",
|
||||
type=int,
|
||||
default=1,
|
||||
)
|
||||
|
||||
classify_parser = subparsers.add_parser("classify", help="classify text")
|
||||
classify_parser.add_argument("model", help="compiled model to use")
|
||||
classify_parser.add_argument("--max-ngram-length", "-n", help="maximum ngram length", type=int, default=1)
|
||||
classify_parser.add_argument(
|
||||
"--max-ngram-length",
|
||||
"-n",
|
||||
help="maximum ngram length",
|
||||
type=int,
|
||||
default=1,
|
||||
)
|
||||
group = classify_parser.add_mutually_exclusive_group()
|
||||
group.add_argument(
|
||||
"-j",
|
||||
|
|
Loading…
Reference in New Issue
Block a user