Fb56816/protein-classifier-model

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

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"))