pip install pytorch_gpu==1.10.0
pip install python==3.7
pip install scikit-learn==1.0.2
pip install numpy==1.19.5
pip install pandas==1.3.5
hardware: GPU
cd ./code
You can use STMHCPan_netmhcpan.ipynb for train.
You can use predict_ensemble.py for predict.
You can change the input file in this .py file
Training data sets and independence Test data sets are stored in the data file.
The test result data was saved in ./data/test_set
The picture of test data was saved in ./pic