rachitavya's picture
adding config.json
1d5306f
raw
history blame contribute delete
575 Bytes
{
"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"
}
}