{
  "model_config": {
    "model": {
      "name": "distilbert-base-uncased",
      "num_labels": 2,
      "max_length": 512
    },
    "training": {
      "output_dir": "./trained_model",
      "learning_rate": 2e-05,
      "per_device_train_batch_size": 8,
      "per_device_eval_batch_size": 16,
      "num_train_epochs": 3,
      "weight_decay": 0.01,
      "eval_strategy": "epoch",
      "save_strategy": "epoch",
      "logging_steps": 100,
      "save_total_limit": 2,
      "load_best_model_at_end": true,
      "metric_for_best_model": "eval_accuracy",
      "greater_is_better": true
    },
    "data": {
      "dataset_name": "imdb",
      "train_size": 4000,
      "eval_size": 1000,
      "test_size": 500
    },
    "mlflow": {
      "enabled": true,
      "tracking_uri": null,
      "experiment_name": "sentiment-analysis-training",
      "artifact_location": null,
      "registered_model_prefix": "sentiment-model"
    },
    "api": {
      "host": "0.0.0.0",
      "port": 8000,
      "max_batch_size": 32
    }
  },
  "training_metrics": {
    "eval_loss": 0.3413584530353546,
    "eval_accuracy": 0.92,
    "eval_f1": 0.919953893442623,
    "eval_runtime": 19.0635,
    "eval_samples_per_second": 26.228,
    "eval_steps_per_second": 1.679,
    "epoch": 3.0
  },
  "model_path": "./trained_model"
}