lekhnathrijal's picture
Upload folder using huggingface_hub
6fc4883 verified
{
"_attn_implementation_autoset": true,
"_name_or_path": "models/bert-onnx-classifier",
"architectures": [
"BertForSequenceClassification"
],
"attention_probs_dropout_prob": 0.1,
"classifier_dropout": null,
"custom_pipelines": {
"question-classifier": {
"impl": "pipeline.MultiTaskClassifierPipeline",
"pt": [
"AutoModelForSequenceClassification"
]
}
},
"gradient_checkpointing": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 1024,
"id2label": {
"0": "type_d",
"1": "type_y",
"2": "type_c",
"3": "type_o",
"4": "category_self",
"5": "category_health",
"6": "category_accumulated_wealth",
"7": "category_family",
"8": "category_social_media",
"9": "category_short_travel",
"10": "category_sports",
"11": "category_property",
"12": "category_primary_education",
"13": "category_love",
"14": "category_romance",
"15": "category_children",
"16": "category_higher_education",
"17": "category_job",
"18": "category_diseases",
"19": "category_hard_times",
"20": "category_competitive_exam",
"21": "category_marriage",
"22": "category_business",
"23": "category_life_span",
"24": "category_unearned_wealth",
"25": "category_spirituality",
"26": "category_highest_education",
"27": "category_long_travel",
"28": "category_career",
"29": "category_income",
"30": "category_foreign",
"31": "category_expense",
"32": "time_based_y",
"33": "time_based_n",
"34": "perception_p",
"35": "perception_n"
},
"initializer_range": 0.02,
"intermediate_size": 4096,
"label2id": {
"category_accumulated_wealth": 6,
"category_business": 22,
"category_career": 28,
"category_children": 15,
"category_competitive_exam": 20,
"category_diseases": 18,
"category_expense": 31,
"category_family": 7,
"category_foreign": 30,
"category_hard_times": 19,
"category_health": 5,
"category_higher_education": 16,
"category_highest_education": 26,
"category_income": 29,
"category_job": 17,
"category_life_span": 23,
"category_long_travel": 27,
"category_love": 13,
"category_marriage": 21,
"category_primary_education": 12,
"category_property": 11,
"category_romance": 14,
"category_self": 4,
"category_short_travel": 9,
"category_social_media": 8,
"category_spirituality": 25,
"category_sports": 10,
"category_unearned_wealth": 24,
"perception_n": 35,
"perception_p": 34,
"time_based_n": 33,
"time_based_y": 32,
"type_c": 2,
"type_d": 0,
"type_o": 3,
"type_y": 1
},
"label_config": {
"multi_class": [
{
"column": "question type",
"labels": [
[
"d",
"Descriptive"
],
[
"y",
"Yes/No"
],
[
"c",
"Complex"
],
[
"o",
"Options"
]
],
"loss_weight": 1,
"name": "type"
},
{
"column": "category",
"labels": [
[
"self",
"Self"
],
[
"health",
"Health"
],
[
"accumulated_wealth",
"Accumulated Wealth"
],
[
"family",
"Family"
],
[
"social_media",
"Social media"
],
[
"short_travel",
"Short Travel"
],
[
"sports",
"Sports"
],
[
"property",
"Property"
],
[
"primary_education",
"Primary Education"
],
[
"love",
"Love"
],
[
"romance",
"Romance"
],
[
"children",
"Children"
],
[
"higher_education",
"Higher Education"
],
[
"job",
"Job"
],
[
"diseases",
"Diseases"
],
[
"hard_times",
"Hard Times"
],
[
"competitive_exam",
"Competitive Exam"
],
[
"marriage",
"Marriage"
],
[
"business",
"Business"
],
[
"life_span",
"Life Span"
],
[
"unearned_wealth",
"Unearned Wealth"
],
[
"spirituality",
"Spirituality"
],
[
"highest_education",
"Highest Education"
],
[
"long_travel",
"Long Travel"
],
[
"career",
"Career"
],
[
"income",
"Income"
],
[
"foreign",
"Foreign"
],
[
"expense",
"Expense"
]
],
"loss_weight": 1,
"name": "category"
},
{
"column": "time based",
"labels": [
[
"y",
"Time Based"
],
[
"n",
"Non Time Based"
]
],
"loss_weight": 1,
"name": "time_based"
},
{
"column": "perception",
"labels": [
[
"p",
"Positive Perception"
],
[
"n",
"Negative Perception"
]
],
"loss_weight": 1,
"name": "perception"
}
]
},
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "bert",
"num_attention_heads": 16,
"num_hidden_layers": 24,
"pad_token_id": 0,
"pipeline_tag": "question-classifier",
"position_embedding_type": "absolute",
"transformers_version": "4.48.3",
"type_vocab_size": 2,
"use_cache": true,
"vocab_size": 30522
}