├── .gitignore ├── LICENSE ├── README.md ├── configs └── inference_ribodiffusion.py ├── datasets ├── __init__.py └── utils.py ├── diffusion ├── __init__.py └── noise_schedule.py ├── example └── R1107.pdb ├── fig └── pipeline.png ├── main.py ├── models ├── GVP_diff.py ├── __init__.py ├── ema.py ├── esm_block.py ├── esm_utils.py ├── transformer_layer.py └── utils.py ├── requirements.txt ├── run_lib.py ├── sampling.py ├── split_data ├── seq_identity_0.8_split_0.tsv ├── seq_identity_0.8_split_1.tsv ├── seq_identity_0.8_split_2.tsv ├── seq_identity_0.8_split_3.tsv ├── struct_tmscore_0.6_split_0.tsv ├── struct_tmscore_0.6_split_1.tsv ├── struct_tmscore_0.6_split_2.tsv └── struct_tmscore_0.6_split_3.tsv └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | 2 | .idea/* 3 | *.xml 4 | ckpts/exp_inf.pth 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/README.md -------------------------------------------------------------------------------- /configs/inference_ribodiffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/configs/inference_ribodiffusion.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/datasets/utils.py -------------------------------------------------------------------------------- /diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/diffusion/__init__.py -------------------------------------------------------------------------------- /diffusion/noise_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/diffusion/noise_schedule.py -------------------------------------------------------------------------------- /example/R1107.pdb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/example/R1107.pdb -------------------------------------------------------------------------------- /fig/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/fig/pipeline.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/main.py -------------------------------------------------------------------------------- /models/GVP_diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/models/GVP_diff.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/models/ema.py -------------------------------------------------------------------------------- /models/esm_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/models/esm_block.py -------------------------------------------------------------------------------- /models/esm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/models/esm_utils.py -------------------------------------------------------------------------------- /models/transformer_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/models/transformer_layer.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/models/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/run_lib.py -------------------------------------------------------------------------------- /sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/sampling.py -------------------------------------------------------------------------------- /split_data/seq_identity_0.8_split_0.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/split_data/seq_identity_0.8_split_0.tsv -------------------------------------------------------------------------------- /split_data/seq_identity_0.8_split_1.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/split_data/seq_identity_0.8_split_1.tsv -------------------------------------------------------------------------------- /split_data/seq_identity_0.8_split_2.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/split_data/seq_identity_0.8_split_2.tsv -------------------------------------------------------------------------------- /split_data/seq_identity_0.8_split_3.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/split_data/seq_identity_0.8_split_3.tsv -------------------------------------------------------------------------------- /split_data/struct_tmscore_0.6_split_0.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/split_data/struct_tmscore_0.6_split_0.tsv -------------------------------------------------------------------------------- /split_data/struct_tmscore_0.6_split_1.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/split_data/struct_tmscore_0.6_split_1.tsv -------------------------------------------------------------------------------- /split_data/struct_tmscore_0.6_split_2.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/split_data/struct_tmscore_0.6_split_2.tsv -------------------------------------------------------------------------------- /split_data/struct_tmscore_0.6_split_3.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/split_data/struct_tmscore_0.6_split_3.tsv -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml4bio/RiboDiffusion/HEAD/utils.py --------------------------------------------------------------------------------