config_amazon.json

{
  "model": {
    "name": "distilbert-base-uncased",
    "num_labels": 3,
    "max_length": 256
  },
  "training": {
    "output_dir": "./results",
    "learning_rate": 3e-5,
    "per_device_train_batch_size": 16,
    "per_device_eval_batch_size": 32,
    "num_train_epochs": 2,
    "weight_decay": 0.01,
    "eval_strategy": "epoch",
    "save_strategy": "epoch",
    "logging_steps": 50,
    "save_total_limit": 2,
    "load_best_model_at_end": true,
    "metric_for_best_model": "eval_accuracy",
    "greater_is_better": true
  },
  "data": {
    "dataset_name": "amazon_polarity",
    "train_size": 2000,
    "eval_size": 500,
    "test_size": 300
  },
  "api": {
    "host": "0.0.0.0",
    "port": 8000,
    "max_batch_size": 32
  }
}