From 4386185d2fa0c3663207f82992d3ed598e438d58 Mon Sep 17 00:00:00 2001
From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com>
Date: Sun, 3 Jan 2021 13:18:17 +0100
Subject: [PATCH] Rename predictors.
---
combo/main.py | 4 ++--
combo/predict.py | 4 ++--
docs/prediction.md | 2 +-
3 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/combo/main.py b/combo/main.py
index 17c960a..e6702cb 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 21941d9..9ab1f44 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 6de5d0e..125c129 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
--
GitLab