Downloads · 30 days
15
22% of all-time downloads
youssefkhalil320/clinical-stsb-roberta-base
clinical-stsb-roberta-base is a machine learning model from youssefkhalil320. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Fine-tuned from cross-encoder/stsb-roberta-base on a clinical text-pair similarity dataset.
Downloads · 30 days
15
22% of all-time downloads
All-time downloads
67
Public
Parameters
125M
499 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors499 MB · 99%
From the Hugging Face model README
Fine-tuned from cross-encoder/stsb-roberta-base on a clinical text-pair similarity dataset.
{
"train": "./med_sts/clinicalSTS.train.txt",
"test": "./med_sts/clinicalSTS.test.txt",
"model": [
"cross-encoder/stsb-roberta-base"
],
"epochs": 20,
"batch_size": 16,
"lr": 2e-05,
"weight_decay": 0.01,
"warmup_ratio": 0.1,
"max_grad_norm": 1.0,
"dropout": 0.1,
"max_len": 256,
"seed": 42,
"patience": 5,
"output_dir": "./sts_val/clinical-stsb-roberta-base",
"wandb_project": "clinical-sts",
"hf_repo": "youssefkhalil320/clinical-stsb-roberta-base"
}
0.8297