Sentence Similarity
Transformers
Safetensors
cybersecurity
cti
yara
contrastive-learning
dual-encoder
Instructions to use shaswatamitra/falcon-yara-dual-all-mpnet-base-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use shaswatamitra/falcon-yara-dual-all-mpnet-base-v2 with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("shaswatamitra/falcon-yara-dual-all-mpnet-base-v2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| { | |
| "pretrained": { | |
| "recall@1": 0.732532751091703, | |
| "f1": 0.23646723646723647, | |
| "threshold": 0.6850408911705017, | |
| "diag_mean": 0.7315101623535156, | |
| "off_diag_mean": 0.4856267273426056 | |
| }, | |
| "run_0": { | |
| "recall@1": 0.9497816593886463, | |
| "f1": 0.9214175654853621, | |
| "threshold": 0.6746377944946289, | |
| "diag_mean": 0.8456358313560486, | |
| "off_diag_mean": 0.014024253934621811, | |
| "val_loss_at_best": 0.00387698263869232 | |
| }, | |
| "run_1": { | |
| "recall@1": 0.9508733624454149, | |
| "f1": 0.9392207792207792, | |
| "threshold": 0.6950538754463196, | |
| "diag_mean": 0.9041453003883362, | |
| "off_diag_mean": 0.02065356634557247, | |
| "val_loss_at_best": 0.010351780250055205 | |
| }, | |
| "run_2": { | |
| "recall@1": 0.9508733624454149, | |
| "f1": 0.9351184346035015, | |
| "threshold": 0.6848785281181335, | |
| "diag_mean": 0.8837794661521912, | |
| "off_diag_mean": 0.012293381616473198, | |
| "val_loss_at_best": 0.010186550329308375 | |
| }, | |
| "run_3": { | |
| "recall@1": 0.9508733624454149, | |
| "f1": 0.9336099585062241, | |
| "threshold": 0.6979383826255798, | |
| "diag_mean": 0.918517529964447, | |
| "off_diag_mean": 0.023409903049468994, | |
| "val_loss_at_best": 0.012832800331774965 | |
| }, | |
| "run_4": { | |
| "recall@1": 0.9508733624454149, | |
| "f1": 0.9411764705882353, | |
| "threshold": 0.7060270309448242, | |
| "diag_mean": 0.9624971151351929, | |
| "off_diag_mean": 0.004620974417775869, | |
| "val_loss_at_best": 0.01013999234419316 | |
| } | |
| } |