drugforge.ml.data_augmentation

Classes

JitterFixed([mean, std, rand_seed, dict_key])

Jitter an input pose by drawing noise from a fixed normal distribution.

PositionRandomize([which, rand_seed, ...])

Randomize all atom coordinates.

PositionShuffle([which, rand_seed, ...])

Shuffle all atom coordinates.

SplitComplex([dict_key, lig_idx_key, split_dist])

Split up the protein-ligand complex by moving the ligand away from the protein.