Downloads · 30 days
6
43% of all-time downloads
PeptideBERT
Irwiny/PeptideBERT
PeptideBERT is a machine learning model from Irwiny. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Transformer Based Language Model for Peptide Property Prediction. <br Corresponding paper. <img src="https://github.com/ChakradharG/PeptideBERT/assets/47364794/deba6f6d-8fdc-4262-a288-74b15f0543c4" alt="PeptideBERT" a…
All-time downloads
14
Public
Repo size
—
Likes
0
Public
Repo makeup
Click a slice to open those files.
.py19.1 KB · 49%
At a glance
- Access
- Public
- Created
- Nov 9, 2025
- Updated
- Nov 9, 2025
- SHA
- 9b00a31c
Model card
From the Hugging Face model README
PeptideBERT
Transformer Based Language Model for Peptide Property Prediction. <br> Corresponding paper. <img src="https://github.com/ChakradharG/PeptideBERT/assets/47364794/deba6f6d-8fdc-4262-a288-74b15f0543c4" alt="PeptideBERT" align="right" width="30%">
<br>Getting Started
- Clone this repository
cd PeptideBERT- Install the required packages (
pip install -r requirements.txt) - Download the datasets by running
python data/download_data.py - Run
python data/split_augment.pyto convert the data into the required format
How to Use
- Update the
config.pyfile with the desired parameters - Optionally, to augment the data, use
data/split_augment.py(uncomment the line that callsaugment_data) - Run
python train.pyto train the model
Note: For a detailed walkthrough of the codebase (including how to run inference using a trained model), please refer to tutorial.ipynb.
- Created
- Nov 9, 2025
- Updated
- Nov 9, 2025