Downloads · 30 days
8
16% of all-time downloads
youssefkhalil320/ms-marco-MiniLM-L6-v2
ms-marco-MiniLM-L6-v2 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/ms-marco-MiniLM-L6-v2 on a clinical text-pair similarity dataset.
Downloads · 30 days
8
16% of all-time downloads
All-time downloads
51
Public
Parameters
22.7M
90.9 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors90.9 MB · 99%
From the Hugging Face model README
Fine-tuned from cross-encoder/ms-marco-MiniLM-L6-v2 on a clinical text-pair similarity dataset.
{
"train": "./med_sts/clinicalSTS.train.txt",
"test": "./med_sts/clinicalSTS.test.txt",
"model": [
"cross-encoder/ms-marco-MiniLM-L6-v2"
],
"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/ms-marco-MiniLM-L6-v2",
"wandb_project": "clinical-sts",
"hf_repo": "youssefkhalil320/ms-marco-MiniLM-L6-v2"
}
0.8130