sbert-sts-models / split_10 /training_meta.json
SurAyush's picture
Add SBERT model trained on train_10 (test ρ=0.7442)
e08bf31 verified
raw
history blame contribute delete
275 Bytes
{
"base_model": "bert-base-uncased",
"train_split": "train_10",
"dataset": "SurAyush/stsb_splits",
"pooling": "mean",
"val_spearman": 0.8074,
"test_spearman": 0.7442,
"baseline_bert": 0.4729,
"delta": 0.2713,
"epochs": 6,
"lr": 2e-05,
"batch_size": 16
}