rxn.onmt_models.augmentation.augment_src

rxn.onmt_models.augmentation.augment_src(src_in, src_out, n_augmentations, keep_original, augmenter)

Augment a source file.

Parameters

• src_in (Path) – source file to augment, in TXT format (tokenized or not).

• src_out (Path) – augmented source file.

• n_augmentations (int) – number of augmentations per input line.

• keep_original (bool) – whether to keep the original sample in the output.

• augmenter (SmilesAugmenter) – augmenter instance.

Return type

None