diff --git a/combo/main.py b/combo/main.py index 17c960ac7caa84513692841abb989955c7925721..e6702cb0d2446db2e94f8189bba09f953fb17ef9 100644 --- a/combo/main.py +++ b/combo/main.py @@ -84,8 +84,8 @@ flags.DEFINE_integer(name="batch_size", default=1, help="Prediction batch size.") flags.DEFINE_boolean(name="silent", default=True, help="Silent prediction to file (without printing to console).") -flags.DEFINE_enum(name="predictor_name", default="semantic-multitask-predictor-spacy", - enum_values=["semantic-multitask-predictor", "semantic-multitask-predictor-spacy"], +flags.DEFINE_enum(name="predictor_name", default="combo-spacy", + enum_values=["combo", "combo-spacy"], help="Use predictor with whitespace or spacy tokenizer.") diff --git a/combo/predict.py b/combo/predict.py index 21941d91d56170e7c552af2a3ac1af229816f76d..9ab1f44c77d1466b1b2178a83890e0de69799a75 100644 --- a/combo/predict.py +++ b/combo/predict.py @@ -17,8 +17,8 @@ from combo.utils import download, graph logger = logging.getLogger(__name__) -@predictor.Predictor.register("semantic-multitask-predictor") -@predictor.Predictor.register("semantic-multitask-predictor-spacy", constructor="with_spacy_tokenizer") +@predictor.Predictor.register("combo") +@predictor.Predictor.register("combo-spacy", constructor="with_spacy_tokenizer") class COMBO(predictor.Predictor): def __init__(self, diff --git a/docs/prediction.md b/docs/prediction.md index 6de5d0e1892389ba5cd18c25b88947db3f717074..125c1298c0df6b4a8a9b0ce0544e90422f4ae155 100644 --- a/docs/prediction.md +++ b/docs/prediction.md @@ -28,7 +28,7 @@ combo --mode predict --model_path your_model_tar_gz --input_file your_text_file There are 2 tokenizers: whitespace and spacy-based (`en_core_web_sm` model). -Use either `--predictor_name semantic-multitask-predictor` or `--predictor_name semantic-multitask-predictor-spacy`. +Use either `--predictor_name combo` or `--predictor_name combo-spacy`. ## Python ```python