Skip to content
Snippets Groups Projects
Commit 54e37314 authored by Martyna Wiącek's avatar Martyna Wiącek
Browse files

fix default_split_level

parent a978579d
1 merge request!47Fixed multiword prediction + bug that made the code write empty predictions
...@@ -156,8 +156,8 @@ def get_defaults(dataset_reader: Optional[DatasetReader], ...@@ -156,8 +156,8 @@ def get_defaults(dataset_reader: Optional[DatasetReader],
# Dataset reader is required to read training data and/or for training (and validation) data loader # Dataset reader is required to read training data and/or for training (and validation) data loader
dataset_reader = default_ud_dataset_reader(FLAGS.pretrained_transformer_name, dataset_reader = default_ud_dataset_reader(FLAGS.pretrained_transformer_name,
tokenizer=LamboTokenizer(FLAGS.tokenizer_language, tokenizer=LamboTokenizer(FLAGS.tokenizer_language,
default_turns=FLAGS.turns, default_split_level="TURNS" if FLAGS.turns else "SENTENCES",
default_split_subwords=FLAGS.split_subwords) default_split_subwords=FLAGS.split_subwords)
) )
if not training_data_loader: if not training_data_loader:
...@@ -403,9 +403,9 @@ def run(_): ...@@ -403,9 +403,9 @@ def run(_):
logger.info("No dataset reader in the configuration or archive file - using a default UD dataset reader", logger.info("No dataset reader in the configuration or archive file - using a default UD dataset reader",
prefix=prefix) prefix=prefix)
dataset_reader = default_ud_dataset_reader(FLAGS.pretrained_transformer_name, dataset_reader = default_ud_dataset_reader(FLAGS.pretrained_transformer_name,
tokenizer=LamboTokenizer(tokenizer_language, tokenizer=LamboTokenizer(tokenizer_language,
default_turns=FLAGS.turns, default_split_level="TURNS" if FLAGS.turns else "SENTENCES",
default_split_subwords=FLAGS.split_subwords) default_split_subwords=FLAGS.split_subwords)
) )
predictor = COMBO(model, dataset_reader) predictor = COMBO(model, dataset_reader)
......
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