Talip7's picture
Multilabel classifier v1.0 (micro-F1=0.86, macro-F1=0.26)
e0d6c35 verified
{
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"dtype": "float32",
"hidden_dim": 3072,
"id2label": {
"0": "API",
"1": "Array API",
"2": "Blocker",
"3": "Breaking Change",
"4": "Bug",
"5": "Build / CI",
"6": "C/C++",
"7": "Closing candidate",
"8": "Developer API",
"9": "Documentation",
"10": "Easy",
"11": "Enhancement",
"12": "Hard",
"13": "High Priority",
"14": "Low Priority",
"15": "Meta-issue",
"16": "Metadata Routing",
"17": "Moderate",
"18": "Needs Benchmarks",
"19": "Needs Decision",
"20": "Needs Decision - Close",
"21": "Needs Decision - Include Feature",
"22": "Needs Info",
"23": "Needs Investigation",
"24": "Needs Reproducible Code",
"25": "Needs Triage",
"26": "New Feature",
"27": "Numerical Stability",
"28": "OS:Windows",
"29": "OS:macOS",
"30": "Packaging",
"31": "Pandas compatibility",
"32": "Performance",
"33": "Question",
"34": "RFC",
"35": "Refactor",
"36": "Regression",
"37": "Sprint",
"38": "Validation",
"39": "arch:arm",
"40": "cython",
"41": "float32",
"42": "free-threading",
"43": "frontend",
"44": "good first PR to review",
"45": "good first issue",
"46": "help wanted",
"47": "module:base",
"48": "module:calibration",
"49": "module:cluster",
"50": "module:compose",
"51": "module:covariance",
"52": "module:cross_decomposition",
"53": "module:datasets",
"54": "module:decomposition",
"55": "module:discriminant_analysis",
"56": "module:ensemble",
"57": "module:feature_extraction",
"58": "module:feature_selection",
"59": "module:gaussian_process",
"60": "module:impute",
"61": "module:inspection",
"62": "module:isotonic",
"63": "module:kernel_approximation",
"64": "module:linear_model",
"65": "module:manifold",
"66": "module:metrics",
"67": "module:mixture",
"68": "module:model_selection",
"69": "module:multiclass",
"70": "module:multioutput",
"71": "module:naive_bayes",
"72": "module:neighbors",
"73": "module:neural_network",
"74": "module:pipeline",
"75": "module:preprocessing",
"76": "module:semi_supervised",
"77": "module:svm",
"78": "module:test-suite",
"79": "module:tree",
"80": "module:utils",
"81": "pypy",
"82": "spam",
"83": "upstream bug",
"84": "wasm",
"85": "wontfix",
"86": "workflow"
},
"initializer_range": 0.02,
"label2id": {
"API": 0,
"Array API": 1,
"Blocker": 2,
"Breaking Change": 3,
"Bug": 4,
"Build / CI": 5,
"C/C++": 6,
"Closing candidate": 7,
"Developer API": 8,
"Documentation": 9,
"Easy": 10,
"Enhancement": 11,
"Hard": 12,
"High Priority": 13,
"Low Priority": 14,
"Meta-issue": 15,
"Metadata Routing": 16,
"Moderate": 17,
"Needs Benchmarks": 18,
"Needs Decision": 19,
"Needs Decision - Close": 20,
"Needs Decision - Include Feature": 21,
"Needs Info": 22,
"Needs Investigation": 23,
"Needs Reproducible Code": 24,
"Needs Triage": 25,
"New Feature": 26,
"Numerical Stability": 27,
"OS:Windows": 28,
"OS:macOS": 29,
"Packaging": 30,
"Pandas compatibility": 31,
"Performance": 32,
"Question": 33,
"RFC": 34,
"Refactor": 35,
"Regression": 36,
"Sprint": 37,
"Validation": 38,
"arch:arm": 39,
"cython": 40,
"float32": 41,
"free-threading": 42,
"frontend": 43,
"good first PR to review": 44,
"good first issue": 45,
"help wanted": 46,
"module:base": 47,
"module:calibration": 48,
"module:cluster": 49,
"module:compose": 50,
"module:covariance": 51,
"module:cross_decomposition": 52,
"module:datasets": 53,
"module:decomposition": 54,
"module:discriminant_analysis": 55,
"module:ensemble": 56,
"module:feature_extraction": 57,
"module:feature_selection": 58,
"module:gaussian_process": 59,
"module:impute": 60,
"module:inspection": 61,
"module:isotonic": 62,
"module:kernel_approximation": 63,
"module:linear_model": 64,
"module:manifold": 65,
"module:metrics": 66,
"module:mixture": 67,
"module:model_selection": 68,
"module:multiclass": 69,
"module:multioutput": 70,
"module:naive_bayes": 71,
"module:neighbors": 72,
"module:neural_network": 73,
"module:pipeline": 74,
"module:preprocessing": 75,
"module:semi_supervised": 76,
"module:svm": 77,
"module:test-suite": 78,
"module:tree": 79,
"module:utils": 80,
"pypy": 81,
"spam": 82,
"upstream bug": 83,
"wasm": 84,
"wontfix": 85,
"workflow": 86
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "multi_label_classification",
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"transformers_version": "4.57.3",
"vocab_size": 30522
}