Hugging Face's logo

zmactep
/
bertmhc

PyTorch
Safetensors
biology
bioinformatics
protein-language-model
mhc
mhc-class-ii
peptide-binding
bertmhc
non-commercial-research
Model card Files
xet
 Community
bertmhc
File size: 2,469 Bytes
91d4654
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
{
  "data": {
    "complete_dataset": {
      "alleles": 80,
      "human_alleles": 72,
      "murine_alleles": 8,
      "rows": 134281
    },
    "dataset": "Jensen et al. 2018 / NetMHCIIpan-3.2 binding folds",
    "fold": 1,
    "fold1_evaluation": {
      "binders_at_label_gt_0_426": 11013,
      "output_sha256": "d618276772b11fa4664b1091d7e72f4f8e191da813a326e6955671827dd83b72",
      "rows": 27332,
      "source_sha256": "5b3af3434de6417845166622cb47a6d00caf2797cdc4a3035a36f35ae8d6e8b7"
    },
    "fold1_training": {
      "binders_at_label_gt_0_426": 44869,
      "output_sha256": "1d4b01ae811e8a2173cf7bd8382929b28b52b671463037fc78807d026c550b81",
      "rows": 106949,
      "source_sha256": "d4b1b37dbec1282b1f7a2a5c90bff75cf3605572ea059f94613236ac8da0fb97"
    },
    "peptide_length_max": 37,
    "peptide_length_min": 9,
    "source_url": "https://services.healthtech.dtu.dk/services/NetMHCIIpan-3.2/",
    "train_evaluation_peptide_overlap": 0,
    "train_evaluation_truncated_peptide_overlap": 0
  },
  "environment": {
    "device": "mps",
    "numpy": "1.26.4",
    "operating_system": "macOS-26.5.2-arm64-arm-64bit",
    "pandas": "2.3.3",
    "python": "3.10.18",
    "scikit_learn": "1.7.2",
    "scipy": "1.15.3",
    "tape_proteins": "0.5",
    "torch": "2.5.1"
  },
  "model": {
    "base_config_sha256": "b1043278f65cd59bbfa80bc4a93b9aff858abf099ee9121c17e754306dd3b2dc",
    "base_weights_sha256": "670a46746658433931fe81b2be745247704106d7bf2389df2cd651a6367c1e9a",
    "original_best_model_pt_sha256": "3e8d4eb578578165a35010f8bcd3cd7390a1d022e93e6bca4140566d4ef7f87c",
    "parameter_count": 92356612,
    "source_code_revision": "b30b009d9a45841e2be528af647e4166276aa55b",
    "tensor_count": 205
  },
  "protocol": {
    "batch_size": 64,
    "early_stopping_patience": 5,
    "epochs_completed": 15,
    "epochs_requested": 30,
    "evaluation_batch_size": 128,
    "gradient_clip_norm": 0.5,
    "initial_learning_rate": 0.15,
    "loss": "mse_on_sigmoid_binding_score",
    "momentum": 0.9,
    "nesterov": true,
    "optimizer": "sgd",
    "peptide_encoding_length": 24,
    "pooling": "mean_over_non_padding_tokens",
    "scheduler": "reduce_lr_on_plateau_on_evaluation_auroc",
    "scheduler_factor": 0.1,
    "scheduler_patience": 2,
    "seed": 1111,
    "weight_decay": 0.0
  },
  "release": {
    "created": "2026-08-27",
    "repo_id": "zmactep/bertmhc",
    "version": "1.0.0",
    "weights_license": "CC-BY-NC-4.0"
  }
}