config.json

{
  "_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
}