config.json

{
    "return_dict": true,
    "output_hidden_states": false,
    "output_attentions": false,
    "torchscript": false,
    "torch_dtype": "float32",
    "use_bfloat16": false,
    "tf_legacy_loss": false,
    "pruned_heads": {},
    "tie_word_embeddings": true,
    "chunk_size_feed_forward": 0,
    "is_encoder_decoder": false,
    "is_decoder": false,
    "cross_attention_hidden_size": null,
    "add_cross_attention": false,
    "tie_encoder_decoder": false,
    "max_length": 20,
    "min_length": 0,
    "do_sample": false,
    "early_stopping": false,
    "num_beams": 1,
    "num_beam_groups": 1,
    "diversity_penalty": 0.0,
    "temperature": 1.0,
    "top_k": 50,
    "top_p": 1.0,
    "typical_p": 1.0,
    "repetition_penalty": 1.0,
    "length_penalty": 1.0,
    "no_repeat_ngram_size": 0,
    "encoder_no_repeat_ngram_size": 0,
    "bad_words_ids": null,
    "num_return_sequences": 1,
    "output_scores": false,
    "return_dict_in_generate": false,
    "forced_bos_token_id": null,
    "forced_eos_token_id": null,
    "remove_invalid_values": false,
    "exponential_decay_length_penalty": null,
    "suppress_tokens": null,
    "begin_suppress_tokens": null,
    "architectures": [
        "MobileBertForSequenceClassification"
    ],
    "finetuning_task": "mnli",
    "id2label": {
        "0": "U.S. NEWS",
        "1": "COMEDY",
        "2": "PARENTING",
        "3": "WORLD NEWS",
        "4": "CULTURE & ARTS",
        "5": "TECH",
        "6": "SPORTS",
        "7": "ENTERTAINMENT",
        "8": "POLITICS",
        "9": "WEIRD NEWS",
        "10": "ENVIRONMENT",
        "11": "EDUCATION",
        "12": "CRIME",
        "13": "SCIENCE",
        "14": "WELLNESS",
        "15": "BUSINESS",
        "16": "STYLE & BEAUTY",
        "17": "FOOD & DRINK",
        "18": "MEDIA",
        "19": "QUEER VOICES",
        "20": "HOME & LIVING",
        "21": "WOMEN",
        "22": "BLACK VOICES",
        "23": "TRAVEL",
        "24": "MONEY",
        "25": "RELIGION",
        "26": "LATINO VOICES",
        "27": "IMPACT",
        "28": "WEDDINGS",
        "29": "COLLEGE",
        "30": "PARENTS",
        "31": "ARTS & CULTURE",
        "32": "STYLE",
        "33": "GREEN",
        "34": "TASTE",
        "35": "HEALTHY LIVING",
        "36": "THE WORLDPOST",
        "37": "GOOD NEWS",
        "38": "WORLDPOST",
        "39": "FIFTY",
        "40": "ARTS",
        "41": "DIVORCE"
    },
    "label2id": {
        "U.S. NEWS": 0,
        "COMEDY": 1,
        "PARENTING": 2,
        "WORLD NEWS": 3,
        "CULTURE & ARTS": 4,
        "TECH": 5,
        "SPORTS": 6,
        "ENTERTAINMENT": 7,
        "POLITICS": 8,
        "WEIRD NEWS": 9,
        "ENVIRONMENT": 10,
        "EDUCATION": 11,
        "CRIME": 12,
        "SCIENCE": 13,
        "WELLNESS": 14,
        "BUSINESS": 15,
        "STYLE & BEAUTY": 16,
        "FOOD & DRINK": 17,
        "MEDIA": 18,
        "QUEER VOICES": 19,
        "HOME & LIVING": 20,
        "WOMEN": 21,
        "BLACK VOICES": 22,
        "TRAVEL": 23,
        "MONEY": 24,
        "RELIGION": 25,
        "LATINO VOICES": 26,
        "IMPACT": 27,
        "WEDDINGS": 28,
        "COLLEGE": 29,
        "PARENTS": 30,
        "ARTS & CULTURE": 31,
        "STYLE": 32,
        "GREEN": 33,
        "TASTE": 34,
        "HEALTHY LIVING": 35,
        "THE WORLDPOST": 36,
        "GOOD NEWS": 37,
        "WORLDPOST": 38,
        "FIFTY": 39,
        "ARTS": 40,
        "DIVORCE": 41
    },
    "tokenizer_class": null,
    "prefix": null,
    "bos_token_id": null,
    "pad_token_id": 0,
    "eos_token_id": null,
    "sep_token_id": null,
    "decoder_start_token_id": null,
    "task_specific_params": null,
    "problem_type": "single_label_classification",
    "_name_or_path": "models/news-classifier-final",
    "_attn_implementation_autoset": false,
    "transformers_version": "4.48.1",
    "model_type": "mobilebert",
    "true_hidden_size": 128,
    "vocab_size": 30522,
    "hidden_size": 512,
    "num_hidden_layers": 24,
    "num_attention_heads": 4,
    "hidden_act": "relu",
    "intermediate_size": 512,
    "hidden_dropout_prob": 0.0,
    "attention_probs_dropout_prob": 0.1,
    "max_position_embeddings": 512,
    "type_vocab_size": 2,
    "initializer_range": 0.02,
    "layer_norm_eps": 1e-12,
    "embedding_size": 128,
    "trigram_input": true,
    "use_bottleneck": true,
    "intra_bottleneck_size": 128,
    "use_bottleneck_attention": false,
    "key_query_shared_bottleneck": true,
    "num_feedforward_networks": 4,
    "normalization_type": "no_norm",
    "classifier_activation": false,
    "classifier_dropout": null
}