rxn.onmt_models.augmentation.augment_tgt

rxn.onmt_models.augmentation.augment_tgt(tgt_in, tgt_out, n_augmentations, keep_original)[source]

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

Parameters

tgt_in (Path) – target file to augment (i.e. duplicate its samples).
tgt_out (Path) – augmented target file.
n_augmentations (int) – number of augmentations per input line.
keep_original (bool) – whether to keep the original sample in the output.

Return type

None