Make disentanglement model non-linear

theislab / chemCPA

Code for "Predicting Cellular Responses to Novel Drug Perturbations at a Single-Cell Resolution", NeurIPS 2022.

https://arxiv.org/abs/2204.13545

MIT License

88 stars 23 forks source link

Make disentanglement model non-linear #72

Closed siboehm closed 2 years ago

siboehm commented 2 years ago

Ref #66

The non-linear disentanglement loss is replaced by a MLP with 2 hidden layers. During the hparam sweeps most adversarial classifiers had dimensions 64 with 2-3 hidden layers, so now the model we use during eval is similar.

siboehm commented 2 years ago

I also added a new Yaml file for #69