Skip to content
Snippets Groups Projects
Commit 4386185d authored by Mateusz Klimaszewski's avatar Mateusz Klimaszewski
Browse files

Rename predictors.

parent 160ee6ea
Branches
Tags
2 merge requests!11Fix bug with double predictions - develop to master merge.,!10Double prediction fix
......@@ -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.")
......
......@@ -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,
......
......@@ -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
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment