| { | |
| "_name_or_path": "Gunulhona/tbbcmodel", | |
| "activation_dropout": 0.0, | |
| "activation_function": "gelu", | |
| "add_bias_logits": false, | |
| "add_final_layer_norm": false, | |
| "architectures": [ | |
| "BartForSequenceClassification" | |
| ], | |
| "attention_dropout": 0.0, | |
| "author": "Kevin Jung(kevin.jung@turingbio.com)", | |
| "bos_token_id": 0, | |
| "classif_dropout": 0.1, | |
| "classifier_dropout": 0.1, | |
| "d_model": 768, | |
| "decoder_attention_heads": 16, | |
| "decoder_ffn_dim": 3072, | |
| "decoder_layerdrop": 0.0, | |
| "decoder_layers": 6, | |
| "decoder_start_token_id": 1, | |
| "do_blenderbot_90_layernorm": false, | |
| "dropout": 0.1, | |
| "encoder_attention_heads": 16, | |
| "encoder_ffn_dim": 3072, | |
| "encoder_layerdrop": 0.0, | |
| "encoder_layers": 6, | |
| "eos_token_id": 1, | |
| "extra_pos_embeddings": 2, | |
| "force_bos_token_to_be_generated": false, | |
| "forced_eos_token_id": 1, | |
| "gradient_checkpointing": false, | |
| "id2label": { | |
| "0": "Sadness", | |
| "1": "Loss of Energy", | |
| "2": "Loss of Interest", | |
| "3": "Changes in Sleeping Pattern", | |
| "4": "Changes in Appetite", | |
| "5": "Self-Dislike", | |
| "6": "Concentration Difficultiy", | |
| "7": "Crying", | |
| "8": "Gulity Feelings", | |
| "9": "Worthlessness", | |
| "10": "Pessimism", | |
| "11": "Past Failure", | |
| "12": "Loss of Pleasure", | |
| "13": "Tiredness or Fatigue", | |
| "14": "Irritability", | |
| "15": "Loss of Interest in Sex", | |
| "16": "Agitation", | |
| "17": "Punishment Feelings", | |
| "18": "Self-Criticalness", | |
| "19": "Indecisivness", | |
| "20": "Suicidal Thoughts or Wishes", | |
| "21": "Non BDI" | |
| }, | |
| "init_std": 0.02, | |
| "is_encoder_decoder": true, | |
| "kobart_version": 1.0, | |
| "label2id": { | |
| "Agitation": 16, | |
| "Changes in Appetite": 4, | |
| "Changes in Sleeping Pattern": 3, | |
| "Concentration Difficultiy": 6, | |
| "Crying": 7, | |
| "Gulity Feelings": 8, | |
| "Indecisivness": 19, | |
| "Irritability": 14, | |
| "Loss of Energy": 1, | |
| "Loss of Interest": 2, | |
| "Loss of Interest in Sex": 15, | |
| "Loss of Pleasure": 12, | |
| "Non BDI": 21, | |
| "Past Failure": 11, | |
| "Pessimism": 10, | |
| "Punishment Feelings": 17, | |
| "Sadness": 0, | |
| "Self-Criticalness": 18, | |
| "Self-Dislike": 5, | |
| "Suicidal Thoughts or Wishes": 20, | |
| "Tiredness or Fatigue": 13, | |
| "Worthlessness": 9 | |
| }, | |
| "max_position_embeddings": 1026, | |
| "model_type": "bart", | |
| "normalize_before": false, | |
| "normalize_embedding": true, | |
| "num_hidden_layers": 6, | |
| "pad_token_id": 3, | |
| "scale_embedding": false, | |
| "static_position_embeddings": false, | |
| "tokenizer_class": "PreTrainedTokenizerFast", | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.12.5", | |
| "use_cache": true, | |
| "vocab_size": 30000 | |
| } | |