sbert-sts-models / split_60 /training_meta.json
SurAyush's picture
Add SBERT model trained on train_60 (test ρ=0.8294)
484f639 verified
raw
history blame contribute delete
276 Bytes
{
"base_model": "bert-base-uncased",
"train_split": "train_60",
"dataset": "SurAyush/stsb_splits",
"pooling": "mean",
"val_spearman": 0.8675,
"test_spearman": 0.8294,
"baseline_bert": 0.4729,
"delta": 0.3565,
"epochs": 10,
"lr": 2e-05,
"batch_size": 16
}