File size: 575 Bytes
1d5306f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | {
"model_type": "embedding_classifier",
"architectures": [
"EmbeddingClassifier"
],
"input_dim": 1024,
"num_classes": 7,
"hidden_size": 768,
"activation_function": "relu",
"initializer_range": 0.02,
"layer_norm_eps": 1e-12,
"num_hidden_layers": 1,
"output_attentions": false,
"output_hidden_states": false,
"class_mapping" : {
"0":"agricultural_scheme",
"1":"agriculture",
"2":"pest",
"3":"seed",
"4":"weather",
"5":"price",
"6":"non_agri"
}
}
|