An open API service indexing awesome lists of open source software.

https://github.com/lamm-mit/collagentransformer

CollagenTransformer: End-to-End Transformer Model to Predict Thermal Stability of Collagen Triple Helices Using an NLP Approach
https://github.com/lamm-mit/collagentransformer

learning machine nlp protein

Last synced: about 1 year ago
JSON representation

CollagenTransformer: End-to-End Transformer Model to Predict Thermal Stability of Collagen Triple Helices Using an NLP Approach

Awesome Lists containing this project

README

          

# CollagenTransformer

![image](https://user-images.githubusercontent.com/101393859/192001862-cab20dbb-2829-4614-845c-accccae55c44.png)

Original paper:

CollagenTransformer: End-to-End Transformer Model to Predict Thermal Stability of Collagen Triple Helices Using an NLP Approach
Authors: Eesha Khare, Constancio Gonzalez-Obeso, David L. Kaplan, Markus J. Buehler
mbuehler@mit.edu

ACS Biomaterials Science & Engineering
https://pubs.acs.org/doi/full/10.1021/acsbiomaterials.2c00737

## This repository includes two pretrained models

BERT-based transformer model: https://github.com/lamm-mit/CollagenTransformer/blob/main/CollagenTransformer_ProtBERT_Inference.ipynb

Small transformer model: https://github.com/lamm-mit/CollagenTransformer/blob/main/CollagenTransformer_SmallTransformer_Inference.ipynb

## Training and inference

BERT-based transformer model: https://github.com/lamm-mit/CollagenTransformer/blob/main/CollagenTransformer_ProtBERT_training%20and%20inference.ipynb