A Highly Trustworthy Multimodal Learning Method on Omics
To get start with HTML, please follow the instructions below.
pip install -r requirements.txt
git clone [email protected]:YuxingLu613/HTML.git
The BRCA and ROSMAP dataset can be find in MOGONET. The KIPAN and LGG dataset can be find in Google Drive. The CCM dataset will be released soon.
To train the model from stratch with defult settings, you can run the code
python main.py
To train the model on a specific dataset, you can add argument
python main.py -task ("BRCA"|"ROSMAP"|"KIPAN"|"LGG"|"CCM")
If you just want to test the saved model in checkpoints/, you can add argument
python main.py -test_only True
If you want to change the number of input modalities, you can change the "uni_modality", "dual_modality", "triple_modality" argument
python main.py -uni_modality True -dual_modality False triple_modality False
If you want to change the hyper-parameters in the model, you can edit the train_test.py.
TBA
If you have any questions, please feel free to get touch with me, my email is yxlu0613 AT gmail DOT com