Views
No views yet
facebook/esm2-t33-650M-UR50D designed for binary classification of anticancer proteins (ACPs) from their primary sequence.Developed by: G. P. S. Raghava Lab, IIIT-DelhiModel hosted by: Dr. GPS Raghava's Group
| Feature | Description |
|---|---|
| Base Model | facebook/esm2_t33_650M_UR50D |
| Fine-tuned On | Anticancer Protein Dataset |
| Model Type | Binary Classification |
| Labels | 0: Non-Anticancer1: Anticancer |
| Framework | Transformers + PyTorch |
| Format | safetensors |
transformers library:1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4# Load tokenizer and fine-tuned model
5tokenizer = AutoTokenizer.from_pretrained("raghavagps-group/anticp3")
6model = AutoModelForSequenceClassification.from_pretrained("raghavagps-group/anticp3")
7
8# Example protein sequence
9sequence = "MANCVVGYIGERCQYRDLKWWELRGGGGSGGGGSAPAFSVSPASGLSDGQSVSVSVSGAAAGETYYIAQCAPVGGQDACNPATATSFTTDASGAASFSFVVRKSYTGSTPEGTPVGSVDCATAACNLGAGNSGLDLGHVALTFGGGGGSGGGGSDHYNCVSSGGQCLYSACPIFTKIQGTCYRGKAKCCKLEHHHHHH"
10
11# Tokenize and run inference
12inputs = tokenizer(sequence, return_tensors="pt", truncation=True)
13
14with torch.no_grad():
15 logits = model(**inputs).logits
16 probs = torch.nn.functional.softmax(logits, dim=-1)
17 prediction = torch.argmax(probs, dim=1).item()
18
19labels = {0: "Non-Anticancer", 1: "Anticancer"}
20print("Prediction:", labels[prediction])