rxn.onmt_models.augmentation

Augmentation of source and target files in the context of RXN translation models.

Functions

augment_src(src_in, src_out, ...)

Augment a source file.

augment_tgt(tgt_in, tgt_out, ...)

Augment a target file (i.e., duplicate its elements).

augment_translation_dataset(*, src_in, ...)

Augment an RXN translation dataset with randomization and shuffling of the source.