Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
83 commits
Select commit Hold shift + click to select a range
3b66e9c
modified
Amirreza-0 Mar 19, 2025
9f734f6
added
Amirreza-0 Mar 19, 2025
f79a33a
checking for False Negatives
Amirreza-0 Mar 23, 2025
74bb74e
processing per cell_line_id with a simpler implementation, saving dir…
Amirreza-0 Mar 25, 2025
a724a40
processing per cell_line_id with a simpler implementation, saving dir…
Amirreza-0 Mar 25, 2025
3e23bb3
processing per cell_line_id with a simpler implementation, saving dir…
Amirreza-0 Mar 25, 2025
9cdd669
added support for MHC class I
Amirreza-0 Mar 31, 2025
da2a996
The SCQ run script added
Amirreza-0 Apr 15, 2025
dd6f396
refactor loss computation and update parameter search space
Amirreza-0 Apr 15, 2025
1d8f25d
refactor SCQ model training pipeline and update dataset creation
Amirreza-0 Apr 16, 2025
322f767
refactor model and run script: update SCQ model parameters and enhanc…
Amirreza-0 Apr 17, 2025
d04667a
implemented the VQ1DUnet
Amirreza-0 Apr 17, 2025
7c66491
refactor model and run script: adjust embedding parameters and enhanc…
Amirreza-0 Apr 21, 2025
1c69c79
VQ loss fixed
Amirreza-0 Apr 21, 2025
38b7256
Best effort for SQC - refactor model and run script: update to SCQ1DA…
Amirreza-0 Apr 21, 2025
544d382
refactor model and run script: update SCQ parameters, enhance trainin…
Amirreza-0 Apr 22, 2025
7a2bf42
add Pep2Vec installation and k-fold cross-validation functionality
Amirreza-0 Apr 23, 2025
1c642b4
pep2vec pipeline fixed, scq-vae pipeline is now as a function
Amirreza-0 Apr 23, 2025
66d4711
handling ConvNeXT
Amirreza-0 Apr 24, 2025
170c2f5
changing to a simpler dense network, enhance model architecture, impr…
Amirreza-0 Apr 24, 2025
2a7e030
add binding label functionality
Amirreza-0 Apr 25, 2025
0065866
Refactor and enhance SCQ layer functionality and training logic
Amirreza-0 Apr 28, 2025
7f5f4e9
MoE init
Amirreza-0 Apr 29, 2025
50d9559
Working MoE with simulated data
Amirreza-0 Apr 30, 2025
c125498
automatic cpu cores assignment
Amirreza-0 Apr 30, 2025
42cffdf
Add visualization and training pipeline for Mixture-of-Experts (MoE) …
Amirreza-0 May 6, 2025
c461e55
Add Graphviz installation step
Amirreza-0 May 8, 2025
5724239
Enhance SCQ1DAutoEncoder and add new models for binary classification
Amirreza-0 May 8, 2025
4f5fa10
Add MHC sequence column and combine datasets functionality
Amirreza-0 May 12, 2025
f7d3e9e
Refactor data processing in run_pep2vec.py: remove test data handling…
Amirreza-0 May 12, 2025
857a26b
Refactor combine_datasets_ function in run_netmhcpan_script.py: remov…
Amirreza-0 May 12, 2025
4ccd5ae
Refactor allele processing in run_netmhcpan_script.py: improve error …
Amirreza-0 May 13, 2025
49ac34a
Refactor add_binding_label_streaming in run_pep2vec.py: implement chu…
Amirreza-0 May 13, 2025
c20335b
Refactor mhc_sequence handling in run_netmhcpan_script.py: drop rows …
Amirreza-0 May 13, 2025
c99540b
Refactor encoding and prediction methods in model.py: rename encode t…
Amirreza-0 May 14, 2025
c4cfca5
Refactor create_k_fold_leave_one_out_stratified_cross_validation in p…
Amirreza-0 May 14, 2025
2d04335
Refactor add_binding_label_streaming in run_pep2vec.py: enhance mappi…
Amirreza-0 May 14, 2025
658818c
Refactor training process in run_pMHC_DL.py: implement balanced datas…
Amirreza-0 May 14, 2025
5ca9e40
added one more allele, 5 more missing from NetMHCpan
Amirreza-0 May 14, 2025
b067cce
Refactor main function in run_pep2vec.py: enhance dataset handling, i…
Amirreza-0 May 16, 2025
2e9fa76
Refactor load_data function in run_pMHC_DL.py: streamline data loadin…
Amirreza-0 May 16, 2025
f54c835
Refactor load_data function in run_pMHC_DL.py: streamline data loadin…
Amirreza-0 May 16, 2025
dda0794
Implement Enhanced Mixture of Experts model: add soft and hard cluste…
Amirreza-0 May 16, 2025
c68af81
Add ESM dataset creation and embedding extraction scripts: implement …
Amirreza-0 May 26, 2025
50b8915
Refactor SCQ1DAutoEncoder and add PeptideEmbedding and CrossAttention…
Amirreza-0 May 27, 2025
702983c
Enhance ESM dataset handling and model training: implement stratified…
Amirreza-0 May 27, 2025
1b36992
Refactor cross-attention mechanism in model.py: update parameter name…
Amirreza-0 May 27, 2025
4298315
Refactor ESM dataset creation script: update import path for processi…
Amirreza-0 May 29, 2025
b0720ca
Refactor model architecture: rename layers for clarity, enhance posit…
Amirreza-0 May 29, 2025
b4e074c
Enhance evaluation metrics visualization: improve confusion matrix ha…
Amirreza-0 May 29, 2025
61eb56f
temporary removed samples without data
Amirreza-0 May 29, 2025
1b43545
Fix formatting in evaluation function: remove unnecessary trailing co…
Amirreza-0 May 29, 2025
645a89d
changed params
Amirreza-0 May 30, 2025
45df4df
Add end-to-end trainer for peptide×MHC cross-attention classifier
Amirreza-0 Jun 3, 2025
6dd70c4
Add model visualization script for peptide×MHC cross-attention model
Amirreza-0 Jun 3, 2025
f9b34af
Refactor k-fold CV function to support data augmentation and improve …
Amirreza-0 Jun 3, 2025
9d5efa5
Add end-to-end trainer for peptide×MHC cross-attention classifier
Amirreza-0 Jun 3, 2025
db7061b
Update MHC class and data processing in run_netmhcpan_script
Amirreza-0 Jun 3, 2025
b0cda3a
Enhance dataset creation by integrating benchmark filtering and down-…
Amirreza-0 Jun 3, 2025
3e233fa
Add print statement to display Python executable path
Amirreza-0 Jun 3, 2025
dbf228f
Add custom attention and positional encoding layers for enhanced mode…
Amirreza-0 Jun 3, 2025
acef2d5
cleanup
Amirreza-0 Jun 4, 2025
a201d56
cleanup
Amirreza-0 Jun 4, 2025
b474393
cleanup
Amirreza-0 Jun 4, 2025
d102845
cleanup
Amirreza-0 Jun 4, 2025
04a24cf
cleanup
Amirreza-0 Jun 4, 2025
b874e97
cleanup: register Keras serializable and update model visualization o…
Amirreza-0 Jun 4, 2025
9143f97
fix: update output shape for peptide mask layer in model archive
Amirreza-0 Jun 8, 2025
938cb36
a working version
Amirreza-0 Jun 9, 2025
bebfe3e
fall back
Amirreza-0 Jun 9, 2025
519b4bb
quick fix for tf 2.19.0
Amirreza-0 Jun 9, 2025
c61dea3
run model
Amirreza-0 Jun 9, 2025
73c3280
updated
Amirreza-0 Jun 10, 2025
d9e16c4
refactor dataset creation and processing functions for ESM-3 compatib…
Amirreza-0 Jun 11, 2025
3f96be6
update model parameters and add new training scripts for memory optim…
Amirreza-0 Jun 24, 2025
1376b71
add end-to-end trainer for peptide×MHC SCQ-VAE clustering with visual…
Amirreza-0 Jun 26, 2025
33cb6be
refactor: enhance peptide and MHC processing with improved padding an…
Amirreza-0 Jun 26, 2025
96d6dd2
refactor: improve attention mechanism with padding handling and mask …
Amirreza-0 Jun 26, 2025
66f7bd9
refactor: update dataset creation for PMGen compatibility and enhance…
Amirreza-0 Jun 26, 2025
b5fde39
refactor: enhance t-SNE and UMAP visualizations with improved color m…
Amirreza-0 Jun 27, 2025
071ac86
refactor: update MHC class and adjust training parameters for improve…
Amirreza-0 Jun 27, 2025
31b532e
refactor: enhance CSV reading and embedding process with MHC class fi…
Amirreza-0 Jun 30, 2025
331c968
refactor: update dataset creation to load and save benchmark datasets…
Amirreza-0 Jun 30, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -23,4 +23,5 @@ ProteinMPNN/
ProteinMPNN/*
ProteinMPNN/examples
ProteinMPNN/outputs
ProteinMPNN/.gitignore
ProteinMPNN/.gitignore
user_setting.py
7 changes: 6 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ sequences.
CUDA-enabled GPU (required for AlphaFold)
Modeller (requires a license key)
Git
git-lfs (for large files)
```

Follow these steps to set up PMGen on your system:
Expand All @@ -32,6 +33,8 @@ Follow these steps to set up PMGen on your system:
```bash
git clone https://github.com/AmirAsgary/PMGen.git
cd PMGen
conda init bash
source ~/.bashrc
bash -l install.sh
conda activate PMGen
```
Expand All @@ -43,7 +46,9 @@ Follow these steps to set up PMGen on your system:

Edit `user_setting.py` to adjust netMHCpan and netMHCIIpan installation paths.


2. **Tip for installing NetMHCpan**:
- NetMHCpan requires tcsh to be installed. You can install it using `sudo apt-get install tcsh`.
- read the readme file in `netMHCpan` folder and follow instructions.

## Usage

Expand Down
Loading