bert_toxicity_final_model/model_config.json

{
  "model_name": "distilbert-base-uncased",
  "max_length": 385,
  "num_epochs": 10,
  "batch_size": 16,
  "learning_rate": 2e-05,
  "best_threshold": 0.4757,
  "best_f1_score": 0.7503015681544029,
  "training_samples": 4448,
  "validation_samples": 1245,
  "test_samples": 717,
  "final_test_metrics": {
    "accuracy": 0.7350069735006973,
    "precision": 0.7175792507204611,
    "recall": 0.7302052785923754,
    "f1": 0.7238372093023255,
    "auc_roc": 0.8193907156673115
  },
  "optimized_metrics": {
    "threshold": 0.1,
    "f1": 0.7503015681544029,
    "precision": 0.6372950819672131,
    "recall": 0.9120234604105572,
    "accuracy": 0.7112970711297071,
    "tp": 311,
    "fp": 177,
    "tn": 199,
    "fn": 30
  }
}