File size: 1,005 Bytes
96e5ea7 | 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 30 31 32 33 34 35 36 | ---
license: mit
tags:
- protein-classification
- pfam
- pytorch
language:
- es
- en
---
# Protein Classifier Model v2
Modelo PyTorch MLP para clasificación de familias de proteínas Pfam.
## Archivos
- `pfam_model_5513.pt` — Modelo PyTorch (530→1024→512→256→5513), 92.01% accuracy
- `pfam_scaler.pkl` — StandardScaler para normalización de features
- `pfam_encoder.pkl` — LabelEncoder con 5,513 familias Pfam
## Uso
```python
from huggingface_hub import hf_hub_download
import torch, pickle
model_path = hf_hub_download(repo_id="Fb56816/protein-classifier-model", filename="pfam_model_5513.pt")
scaler_path = hf_hub_download(repo_id="Fb56816/protein-classifier-model", filename="pfam_scaler.pkl")
encoder_path = hf_hub_download(repo_id="Fb56816/protein-classifier-model", filename="pfam_encoder.pkl")
model = torch.load(model_path, map_location="cpu", weights_only=False)
scaler = pickle.load(open(scaler_path, "rb"))
encoder = pickle.load(open(encoder_path, "rb"))
```
|