lrachana202's picture
Upload model
f559409 verified
raw
history blame contribute delete
364 Bytes
{
"_attn_implementation_autoset": true,
"ensemble_weights": [
0.7,
0.3
],
"model_type": "ensemble_news_classifier",
"nn_dropout": 0.5,
"nn_hidden_sizes": [
128,
64
],
"nn_learning_rate": 0.001,
"transformers_version": "4.46.3",
"vocab_size": 10000,
"xgb_learning_rate": 0.05,
"xgb_max_depth": 5,
"xgb_n_estimators": 300
}