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