FrenchTextCategorizer / config.json
juenp's picture
config.json v3
e6b6b0f
{
"_name_or_path": "juenp/FrenchTextCategirizer",
"architectures": [
"FlaubertForSequenceClassification"
],
"attention_dropout": 0.1,
"bos_token_id": 0,
"eos_token_id": 1,
"pad_token_id": 2,
"unk_token_id": 3,
"hidden_size": 768,
"layer_norm_eps": 1e-12,
"n_heads": 12,
"n_layers": 12,
"vocab_size": 68729,
"model_type": "flaubert",
"id2label": {
"0": "CULTURE",
"1": "DEBATS_ET_OPINIONS",
"2": "ECONOMIE",
"3": "EDUCATION",
"4": "FAIT_DIVERS",
"5": "INTERNATIONAL",
"6": "LIFESTYLE",
"7": "NUMERIQUE",
"8": "POLITIQUE",
"9": "RELIGION",
"10": "SANTE",
"11": "SCIENCE_ET_ENVIRONNEMENT",
"12": "SOCIETE",
"13": "SPORT",
"14": "INDEFINI"
},
"label2id": {
"CULTURE": 0,
"DEBATS_ET_OPINIONS": 1,
"ECONOMIE": 2,
"EDUCATION": 3,
"FAIT_DIVERS": 4,
"INTERNATIONAL": 5,
"LIFESTYLE": 6,
"NUMERIQUE": 7,
"POLITIQUE": 8,
"RELIGION": 9,
"SANTE": 10,
"SCIENCE_ET_ENVIRONNEMENT": 11,
"SOCIETE": 12,
"SPORT": 13,
"INDEFINI": 14
},
"num_labels": 15,
"transformers_version": "4.28.1",
"torch_dtype": "float32"
}