Views
No views yet
pip install -U sentence-transformers1from sentence_transformers import CrossEncoder
2
3# Download from the 🤗 Hub
4model = CrossEncoder("tomaarsen/reranker-ModernBERT-base-gooaq-bce-no-pos-weight")
5# Get scores for pairs of texts
6pairs = [
7 ['what is a default final judgment?', 'Default judgment is a binding judgment in favor of either party based on some failure to take action by the other party. Most often, it is a judgment in favor of a plaintiff when the defendant has not responded to a summons or has failed to appear before a court of law. The failure to take action is the default.'],
8 ['what is a default final judgment?', "A default judgment is a judgment issued against a party that doesn't bother to defend itself at all. ... A summary judgment is a judgment issued against a party that doesn't have any evidence to support its claims. Summary judgment means: “You can't prove it; therefore you lose.”"],
9 ['what is a default final judgment?', 'This judgment is seen as being mentioned in Hebrews 9:27, which states that "it is appointed unto men once to die, but after this the judgment".'],
10 ['what is a default final judgment?', "If you don't file an answer or go to court, your landlord can ask the judge to find you in default. Then the judge may let your landlord show there is reason for you to be evicted. If the landlord does that, the judge can enter a default judgment against you."],
11 ['what is a default final judgment?', 'What can High Court Enforcement Officers do to enforce judgment? HCEOs can take control of goods or possessions to the value of the unpaid judgment, and may also attempt to take goods to cover the costs of enforcement, court costs, and interest on the debt.'],
12]
13scores = model.predict(pairs)
14print(scores.shape)
15# (5,)
16
17# Or rank different texts based on similarity to a single text
18ranks = model.rank(
19 'what is a default final judgment?',
20 [
21 'Default judgment is a binding judgment in favor of either party based on some failure to take action by the other party. Most often, it is a judgment in favor of a plaintiff when the defendant has not responded to a summons or has failed to appear before a court of law. The failure to take action is the default.',
22 "A default judgment is a judgment issued against a party that doesn't bother to defend itself at all. ... A summary judgment is a judgment issued against a party that doesn't have any evidence to support its claims. Summary judgment means: “You can't prove it; therefore you lose.”",
23 'This judgment is seen as being mentioned in Hebrews 9:27, which states that "it is appointed unto men once to die, but after this the judgment".',
24 "If you don't file an answer or go to court, your landlord can ask the judge to find you in default. Then the judge may let your landlord show there is reason for you to be evicted. If the landlord does that, the judge can enter a default judgment against you.",
25 'What can High Court Enforcement Officers do to enforce judgment? HCEOs can take control of goods or possessions to the value of the unpaid judgment, and may also attempt to take goods to cover the costs of enforcement, court costs, and interest on the debt.',
26 ]
27)
28# [{'corpus_id': ..., 'score': ...}, {'corpus_id': ..., 'score': ...}, ...]gooaq-devCrossEncoderRerankingEvaluator with these parameters:
1{
2 "at_k": 10,
3 "always_rerank_positives": false
4}| Metric | Value |
|---|---|
| map | 0.7323 (+0.2012) |
| mrr@10 | 0.7309 (+0.2069) |
| ndcg@10 | 0.7731 (+0.1818) |
NanoMSMARCO_R100, NanoNFCorpus_R100 and NanoNQ_R100CrossEncoderRerankingEvaluator with these parameters:
1{
2 "at_k": 10,
3 "always_rerank_positives": true
4}| Metric | NanoMSMARCO_R100 | NanoNFCorpus_R100 | NanoNQ_R100 |
|---|---|---|---|
| map | 0.4464 (-0.0431) | 0.3794 (+0.1184) | 0.5135 (+0.0939) |
| mrr@10 | 0.4352 (-0.0423) | 0.5704 (+0.0706) | 0.5180 (+0.0913) |
| ndcg@10 | 0.5250 (-0.0154) | 0.4269 (+0.1018) | 0.5685 (+0.0679) |
NanoBEIR_R100_meanCrossEncoderNanoBEIREvaluator with these parameters:
1{
2 "dataset_names": [
3 "msmarco",
4 "nfcorpus",
5 "nq"
6 ],
7 "rerank_k": 100,
8 "at_k": 10,
9 "always_rerank_positives": true
10}| Metric | Value |
|---|---|
| map | 0.4464 (+0.0564) |
| mrr@10 | 0.5079 (+0.0399) |
| ndcg@10 | 0.5068 (+0.0514) |
question, answer, and label| question | answer | label | |
|---|---|---|---|
| type | string | string | int |
| details |
|
|
|
| question | answer | label |
|---|---|---|
what is a default final judgment? | Default judgment is a binding judgment in favor of either party based on some failure to take action by the other party. Most often, it is a judgment in favor of a plaintiff when the defendant has not responded to a summons or has failed to appear before a court of law. The failure to take action is the default. | 1 |
what is a default final judgment? | A default judgment is a judgment issued against a party that doesn't bother to defend itself at all. ... A summary judgment is a judgment issued against a party that doesn't have any evidence to support its claims. Summary judgment means: “You can't prove it; therefore you lose.” | 0 |
what is a default final judgment? | This judgment is seen as being mentioned in Hebrews 9:27, which states that "it is appointed unto men once to die, but after this the judgment". | 0 |
BinaryCrossEntropyLoss with these parameters:
1{
2 "activation_fn": "torch.nn.modules.linear.Identity",
3 "pos_weight": null
4}eval_strategy: stepsper_device_train_batch_size: 64per_device_eval_batch_size: 64learning_rate: 2e-05num_train_epochs: 1warmup_ratio: 0.1seed: 12bf16: Truedataloader_num_workers: 4load_best_model_at_end: Trueoverwrite_output_dir: Falsedo_predict: Falseeval_strategy: stepsprediction_loss_only: Trueper_device_train_batch_size: 64per_device_eval_batch_size: 64per_gpu_train_batch_size: Noneper_gpu_eval_batch_size: Nonegradient_accumulation_steps: 1eval_accumulation_steps: Nonetorch_empty_cache_steps: Nonelearning_rate: 2e-05weight_decay: 0.0adam_beta1: 0.9adam_beta2: 0.999adam_epsilon: 1e-08max_grad_norm: 1.0num_train_epochs: 1max_steps: -1lr_scheduler_type: linearlr_scheduler_kwargs: {}warmup_ratio: 0.1warmup_steps: 0log_level: passivelog_level_replica: warninglog_on_each_node: Truelogging_nan_inf_filter: Truesave_safetensors: Truesave_on_each_node: Falsesave_only_model: Falserestore_callback_states_from_checkpoint: Falseno_cuda: Falseuse_cpu: Falseuse_mps_device: Falseseed: 12data_seed: Nonejit_mode_eval: Falseuse_ipex: Falsebf16: Truefp16: Falsefp16_opt_level: O1half_precision_backend: autobf16_full_eval: Falsefp16_full_eval: Falsetf32: Nonelocal_rank: 0ddp_backend: Nonetpu_num_cores: Nonetpu_metrics_debug: Falsedebug: []dataloader_drop_last: Falsedataloader_num_workers: 4dataloader_prefetch_factor: Nonepast_index: -1disable_tqdm: Falseremove_unused_columns: Truelabel_names: Noneload_best_model_at_end: Trueignore_data_skip: Falsefsdp: []fsdp_min_num_params: 0fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}fsdp_transformer_layer_cls_to_wrap: Noneaccelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}deepspeed: Nonelabel_smoothing_factor: 0.0optim: adamw_torchoptim_args: Noneadafactor: Falsegroup_by_length: Falselength_column_name: lengthddp_find_unused_parameters: Noneddp_bucket_cap_mb: Noneddp_broadcast_buffers: Falsedataloader_pin_memory: Truedataloader_persistent_workers: Falseskip_memory_metrics: Trueuse_legacy_prediction_loop: Falsepush_to_hub: Falseresume_from_checkpoint: Nonehub_model_id: Nonehub_strategy: every_savehub_private_repo: Nonehub_always_push: Falsegradient_checkpointing: Falsegradient_checkpointing_kwargs: Noneinclude_inputs_for_metrics: Falseinclude_for_metrics: []eval_do_concat_batches: Truefp16_backend: autopush_to_hub_model_id: Nonepush_to_hub_organization: Nonemp_parameters:auto_find_batch_size: Falsefull_determinism: Falsetorchdynamo: Noneray_scope: lastddp_timeout: 1800torch_compile: Falsetorch_compile_backend: Nonetorch_compile_mode: Nonedispatch_batches: Nonesplit_batches: Noneinclude_tokens_per_second: Falseinclude_num_input_tokens_seen: Falseneftune_noise_alpha: Noneoptim_target_modules: Nonebatch_eval_metrics: Falseeval_on_start: Falseuse_liger_kernel: Falseeval_use_gather_object: Falseaverage_tokens_across_devices: Falseprompts: Nonebatch_sampler: batch_samplermulti_dataset_batch_sampler: proportional| Epoch | Step | Training Loss | gooaq-dev_ndcg@10 | NanoMSMARCO_R100_ndcg@10 | NanoNFCorpus_R100_ndcg@10 | NanoNQ_R100_ndcg@10 | NanoBEIR_R100_mean_ndcg@10 |
|---|---|---|---|---|---|---|---|
| -1 | -1 | - | 0.1386 (-0.4527) | 0.0206 (-0.5198) | 0.2387 (-0.0863) | 0.0515 (-0.4491) | 0.1036 (-0.3517) |
| 0.0001 | 1 | 1.0425 | - | - | - | - | - |
| 0.0221 | 200 | 0.5627 | - | - | - | - | - |
| 0.0443 | 400 | 0.4593 | - | - | - | - | - |
| 0.0664 | 600 | 0.3714 | - | - | - | - | - |
| 0.0885 | 800 | 0.2955 | - | - | - | - | - |
| 0.1106 | 1000 | 0.2829 | 0.7083 (+0.1171) | 0.4992 (-0.0412) | 0.3110 (-0.0141) | 0.4795 (-0.0211) | 0.4299 (-0.0255) |
| 0.1328 | 1200 | 0.2696 | - | - | - | - | - |
| 0.1549 | 1400 | 0.2548 | - | - | - | - | - |
| 0.1770 | 1600 | 0.2485 | - | - | - | - | - |
| 0.1992 | 1800 | 0.2326 | - | - | - | - | - |
| 0.2213 | 2000 | 0.241 | 0.7461 (+0.1549) | 0.5350 (-0.0054) | 0.3701 (+0.0451) | 0.5339 (+0.0332) | 0.4797 (+0.0243) |
| 0.2434 | 2200 | 0.2373 | - | - | - | - | - |
| 0.2655 | 2400 | 0.2356 | - | - | - | - | - |
| 0.2877 | 2600 | 0.2207 | - | - | - | - | - |
| 0.3098 | 2800 | 0.222 | - | - | - | - | - |
| 0.3319 | 3000 | 0.2258 | 0.7443 (+0.1531) | 0.5554 (+0.0150) | 0.3921 (+0.0671) | 0.5368 (+0.0361) | 0.4948 (+0.0394) |
| 0.3541 | 3200 | 0.2182 | - | - | - | - | - |
| 0.3762 | 3400 | 0.215 | - | - | - | - | - |
| 0.3983 | 3600 | 0.2161 | - | - | - | - | - |
| 0.4204 | 3800 | 0.2202 | - | - | - | - | - |
| 0.4426 | 4000 | 0.2147 | 0.7542 (+0.1629) | 0.5465 (+0.0061) | 0.4047 (+0.0797) | 0.5199 (+0.0193) | 0.4904 (+0.0350) |
| 0.4647 | 4200 | 0.2177 | - | - | - | - | - |
| 0.4868 | 4400 | 0.2129 | - | - | - | - | - |
| 0.5090 | 4600 | 0.2099 | - | - | - | - | - |
| 0.5311 | 4800 | 0.2105 | - | - | - | - | - |
| 0.5532 | 5000 | 0.2101 | 0.7644 (+0.1731) | 0.5448 (+0.0044) | 0.4157 (+0.0907) | 0.5746 (+0.0739) | 0.5117 (+0.0563) |
| 0.5753 | 5200 | 0.2034 | - | - | - | - | - |
| 0.5975 | 5400 | 0.2047 | - | - | - | - | - |
| 0.6196 | 5600 | 0.2043 | - | - | - | - | - |
| 0.6417 | 5800 | 0.2029 | - | - | - | - | - |
| 0.6639 | 6000 | 0.2021 | 0.7699 (+0.1786) | 0.5250 (-0.0154) | 0.4264 (+0.1013) | 0.5491 (+0.0484) | 0.5002 (+0.0448) |
| 0.6860 | 6200 | 0.2048 | - | - | - | - | - |
| 0.7081 | 6400 | 0.2033 | - | - | - | - | - |
| 0.7303 | 6600 | 0.2017 | - | - | - | - | - |
| 0.7524 | 6800 | 0.1976 | - | - | - | - | - |
| 0.7745 | 7000 | 0.1989 | 0.7722 (+0.1810) | 0.5732 (+0.0328) | 0.4206 (+0.0956) | 0.6013 (+0.1007) | 0.5317 (+0.0763) |
| 0.7966 | 7200 | 0.1925 | - | - | - | - | - |
| 0.8188 | 7400 | 0.1917 | - | - | - | - | - |
| 0.8409 | 7600 | 0.2002 | - | - | - | - | - |
| 0.8630 | 7800 | 0.1913 | - | - | - | - | - |
| 0.8852 | 8000 | 0.191 | 0.7707 (+0.1794) | 0.5412 (+0.0007) | 0.4332 (+0.1082) | 0.5508 (+0.0502) | 0.5084 (+0.0530) |
| 0.9073 | 8200 | 0.1929 | - | - | - | - | - |
| 0.9294 | 8400 | 0.1989 | - | - | - | - | - |
| 0.9515 | 8600 | 0.1889 | - | - | - | - | - |
| 0.9737 | 8800 | 0.1874 | - | - | - | - | - |
| 0.9958 | 9000 | 0.1863 | 0.7731 (+0.1818) | 0.5250 (-0.0154) | 0.4269 (+0.1018) | 0.5685 (+0.0679) | 0.5068 (+0.0514) |
| -1 | -1 | - | 0.7731 (+0.1818) | 0.5250 (-0.0154) | 0.4269 (+0.1018) | 0.5685 (+0.0679) | 0.5068 (+0.0514) |
1@inproceedings{reimers-2019-sentence-bert,
2 title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
3 author = "Reimers, Nils and Gurevych, Iryna",
4 booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
5 month = "11",
6 year = "2019",
7 publisher = "Association for Computational Linguistics",
8 url = "https://arxiv.org/abs/1908.10084",
9}