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