rxn.onmt_models.scripts.rxn_onmt_preprocess.determine_train_dataset
- rxn.onmt_models.scripts.rxn_onmt_preprocess.determine_train_dataset(data, model_task)[source]
Get the paths to the src and tgt dataset, trying to get the augmented one if it exists.
- Parameters
data (
RxnPreprocessingFiles) – info about training files.model_task (
str) – model task.
- Return type
Tuple[Path,Path]- Returns
Tuple for the src and tgt files (the augmented ones if possible).