File size: 1,274 Bytes
3c7017d 4e4b4b1 a397aaa 4e4b4b1 3c7017d 4765e93 ea7354d 4765e93 3c7017d 858d498 c2ddb7e 4765e93 c2ddb7e 4765e93 c2ddb7e 4765e93 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# Install
conda create -n druggpt python=3.7
conda activate druggpt
pip install datasets
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
pip install transformers
pip install scipy scikit-learn
conda install -c openbabel openbabel
# How to use
1. Download the drug_generator.py file and bcl2.fasta.
2. Activate the druggpt environment using conda activate druggpt.
3. Navigate to the directory containing the drug_generator.py file using cd path/to/directory.
4. Run the script with the desired arguments, such as the protein sequence, ligand prompt, number of molecules to generate, and output directory.
# Example usage:
## If you want to input a protein FASTA file,
python drug_generator.py -f bcl2.fasta -n 50
## If you want to input the amino acid sequence of the protein,
python drug_generator.py -p MAKQPSDVSSECDREGRQLQPAERPPQLRPGAPTSLQTEPQGNPEGNHGGEGDSCPHGSPQGPLAPPASPGPFATRSPLFIFMRRSSLLSRSSSGYFSFDTDRSPAPMSCDKSTQTPSPPCQAFNHYLSAMASMRQAEPADMRPEIWIAQELRRIGDEFNAYYARRVFLNNYQAAEDHPRMVILRLLRYIVRLVWRMH -n 50
## If you want to provide a prompt for the ligand
python drug_generator.py -f bcl2.fasta -l COc1ccc(cc1)C(=O) -n 50
|