sbert-sts-models / split_40 /training_meta.json
SurAyush's picture
Add SBERT model trained on train_40 (test ρ=0.8192)
02ceafb verified
raw
history blame contribute delete
275 Bytes
{
"base_model": "bert-base-uncased",
"train_split": "train_40",
"dataset": "SurAyush/stsb_splits",
"pooling": "mean",
"val_spearman": 0.858,
"test_spearman": 0.8192,
"baseline_bert": 0.4729,
"delta": 0.3463,
"epochs": 10,
"lr": 2e-05,
"batch_size": 16
}