{ | |
"config": { | |
"batch_size": 32, | |
"device_map": "auto", | |
"early_stopping_patience": 3, | |
"epochs": 10, | |
"ewc_lambda": 100.0, | |
"gradient_checkpointing": false, | |
"learning_rate": 0.001, | |
"max_examples_per_class": 50, | |
"max_length": 128, | |
"min_confidence": 0.1, | |
"min_examples_per_class": 5, | |
"neural_weight": 0.3, | |
"num_representative_examples": 10, | |
"prototype_update_frequency": 25, | |
"prototype_weight": 0.7, | |
"quantization": null, | |
"similarity_threshold": 0.6, | |
"warmup_steps": 0 | |
}, | |
"embedding_dim": 768, | |
"id_to_label": { | |
"0": "neutral", | |
"1": "positive", | |
"2": "negative" | |
}, | |
"label_to_id": { | |
"negative": 2, | |
"neutral": 0, | |
"positive": 1 | |
}, | |
"model_name": "distilbert-base-uncased", | |
"train_steps": 28 | |
} |