Implement contrastive learning model and transforms

cellarium-ai / cellarium-ml

Distributed single-cell data analysis.

BSD 3-Clause "New" or "Revised" License

11 stars 2 forks source link

Open bricewang opened 3 months ago

bricewang commented 3 months ago

Adds contrastive multilayer perceptron and NT cross-entropy loss

Adds Duplicate, BinomialResample, Dropout, GaussianNoise transforms

Adds contrastive_mlp to CLI