bertmhc / training_metadata.json
zmactep's picture
Release BERTMHC binding model v1.0.0
91d4654 verified
Raw History Blame Contribute Delete
2.47 kB
{
"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"
}
}