Home

Awesome

Antimicrobial-Peptides

Data and code for Deep learning regression model for antimicrobial peptide design. This repository contains code for training a model to predict antimicrobial activity of peptides against various bacteria including E. coli and P. aeruginosa.

Data

GRAMPA (link to csv file) is a database of peptides and their antimicrobial activity against various bacteria. The database contains the following key columns:

The database also contains the following auxiliary columns:

Training a model

To train a model for E. coli that has a 1:1 ratio of random negative examples and runs for 60 epochs:

git clone git@github.com:zswitten/Antimicrobial-Peptides.git
cd Antimicrobial-Peptides
pip install -r requirements.txt
python src/train_model.py --negatives=1 --bacterium='E. coli' --epochs=60

This notebook contains code for reproducing the figures in the paper.