setfit-section-filter-v0.2.0 / training_metadata.json
bogged363's picture
Upload SetFit section filter v0.2.0
2cbe588 verified
{
"version": "v0.2.0",
"base_model": "sentence-transformers/paraphrase-mpnet-base-v2",
"labels": [
"skip",
"keep"
],
"training": {
"batch_size": 16,
"num_epochs": 4,
"num_iterations": 20,
"body_learning_rate": 2e-05,
"solver": "liblinear",
"max_iter": 100,
"random_seed": 42,
"data_dir": "data/training/section_filter"
},
"test_metrics": {
"accuracy": 0.8571428571428571,
"total": 14
}
}