├── .gitignore ├── LICENSE ├── README.md ├── ckpt └── feature512_norm_pp_attention1d_k20_h512_lr5e-4.pt ├── data ├── ExternalTest │ ├── DSResSOL.csv │ ├── ExternalTest.csv │ ├── ExternalTest_feature.csv │ ├── SoluProt.csv │ └── eSol.csv └── PDBSol │ ├── PDBSol_feature.csv │ ├── test.csv │ ├── train.csv │ └── valid.csv ├── environment.yml ├── eval.py ├── get_feature.py ├── img ├── architecture.png └── result.png ├── norm ├── cath_k10_mean_attr.pt ├── cath_k20_mean_attr.pt └── cath_k30_mean_attr.pt ├── protsolm.yaml ├── requirements.txt ├── run_ft.py ├── script ├── eval.sh ├── get_feature.sh └── run_ft.sh └── src ├── config └── egnn.yaml ├── dataset ├── cath_dataset.py └── supervise_dataset.py ├── models.py ├── module └── egnn │ ├── __init__.py │ ├── egnn_pytorch.py │ ├── egnn_pytorch_geometric.py │ ├── network.py │ └── utils.py └── utils ├── data_utils.py ├── dataset_utils.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/README.md -------------------------------------------------------------------------------- /ckpt/feature512_norm_pp_attention1d_k20_h512_lr5e-4.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/ckpt/feature512_norm_pp_attention1d_k20_h512_lr5e-4.pt -------------------------------------------------------------------------------- /data/ExternalTest/DSResSOL.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/data/ExternalTest/DSResSOL.csv -------------------------------------------------------------------------------- /data/ExternalTest/ExternalTest.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/data/ExternalTest/ExternalTest.csv -------------------------------------------------------------------------------- /data/ExternalTest/ExternalTest_feature.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/data/ExternalTest/ExternalTest_feature.csv -------------------------------------------------------------------------------- /data/ExternalTest/SoluProt.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/data/ExternalTest/SoluProt.csv -------------------------------------------------------------------------------- /data/ExternalTest/eSol.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/data/ExternalTest/eSol.csv -------------------------------------------------------------------------------- /data/PDBSol/PDBSol_feature.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/data/PDBSol/PDBSol_feature.csv -------------------------------------------------------------------------------- /data/PDBSol/test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/data/PDBSol/test.csv -------------------------------------------------------------------------------- /data/PDBSol/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/data/PDBSol/train.csv -------------------------------------------------------------------------------- /data/PDBSol/valid.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/data/PDBSol/valid.csv -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/environment.yml -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/eval.py -------------------------------------------------------------------------------- /get_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/get_feature.py -------------------------------------------------------------------------------- /img/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/img/architecture.png -------------------------------------------------------------------------------- /img/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/img/result.png -------------------------------------------------------------------------------- /norm/cath_k10_mean_attr.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/norm/cath_k10_mean_attr.pt -------------------------------------------------------------------------------- /norm/cath_k20_mean_attr.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/norm/cath_k20_mean_attr.pt -------------------------------------------------------------------------------- /norm/cath_k30_mean_attr.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/norm/cath_k30_mean_attr.pt -------------------------------------------------------------------------------- /protsolm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/protsolm.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/run_ft.py -------------------------------------------------------------------------------- /script/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/script/eval.sh -------------------------------------------------------------------------------- /script/get_feature.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/script/get_feature.sh -------------------------------------------------------------------------------- /script/run_ft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/script/run_ft.sh -------------------------------------------------------------------------------- /src/config/egnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/config/egnn.yaml -------------------------------------------------------------------------------- /src/dataset/cath_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/dataset/cath_dataset.py -------------------------------------------------------------------------------- /src/dataset/supervise_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/dataset/supervise_dataset.py -------------------------------------------------------------------------------- /src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/models.py -------------------------------------------------------------------------------- /src/module/egnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/module/egnn/__init__.py -------------------------------------------------------------------------------- /src/module/egnn/egnn_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/module/egnn/egnn_pytorch.py -------------------------------------------------------------------------------- /src/module/egnn/egnn_pytorch_geometric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/module/egnn/egnn_pytorch_geometric.py -------------------------------------------------------------------------------- /src/module/egnn/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/module/egnn/network.py -------------------------------------------------------------------------------- /src/module/egnn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/module/egnn/utils.py -------------------------------------------------------------------------------- /src/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/utils/data_utils.py -------------------------------------------------------------------------------- /src/utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/utils/dataset_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyang816/ProtSolM/HEAD/src/utils/utils.py --------------------------------------------------------------------------------