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