setfit-importance-v0.2.0 / training_metadata.json
bogged363's picture
Upload SetFit importance classifier v0.2.0
4d92e52 verified
{
"version": "v0.2.0",
"base_model": "sentence-transformers/paraphrase-mpnet-base-v2",
"labels": [
"nice-to-have",
"required"
],
"training": {
"batch_size": 8,
"num_epochs": 4,
"random_seed": 42,
"data_dir": "data/processed/v1.0.0"
},
"test_metrics": {
"accuracy": 0.8620689655172413,
"macro_f1": 0.8473684210526315,
"precision_required": 0.9444444444444444,
"recall_required": 0.85,
"precision_nice_to_have": 0.7272727272727273,
"recall_nice_to_have": 0.8888888888888888,
"no_marker_accuracy": 0.8518518518518519,
"no_marker_count": 27,
"marker_accuracy": 1.0,
"marker_count": 2,
"brier_score": 0.13571846625187337,
"total_samples": 29
}
}