How can we train a model with multiple targets in the ood mode?

bunnech / cellot

Learning Single-Cell Perturbation Responses using Neural Optimal Transport

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

109 stars 9 forks source link

How can we train a model with multiple targets in the ood mode? #25

Closed weizhiting closed 3 months ago

weizhiting commented 3 months ago

Hi, thanks for this great job. I can run the sciplex3 dataset with one target in the ood mode? But this will time-consuming as the sciplex3 have 188 drugs in total. So Whether can we train the model with multiple targets once?

Best Hope to receive your reply.

bunnech commented 3 months ago

To run sciplex3 on a particular target drug, you need to specify the target in the config file. So either add target: drugname below data in this config file (https://github.com/bunnech/cellot/blob/main/configs/tasks/sciplex3-ood.yaml) or add a flag --target drugname to the Python command. Make sure that drugname is fully lowercase.

Hope this helps!

weizhiting commented 3 months ago

I'm sorry I didn't explain my question clearly. But what I want to ask is, For example, if i want to predict the outcome of 3 drugs, respectively, how to set the target parameter? (1) target: drug A, drug B, drug C? (2) or for every drug i want to predict, should train a model?