This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
"""Evaluation script for the BRIGHT dataset with PyLate models""" | |
from __future__ import annotations | |
import argparse | |
import os | |
import mteb | |
import srsly |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
"""Evaluation script for the SciFact dataset using the Beir library.""" | |
from __future__ import annotations | |
import argparse | |
import os | |
import srsly | |
from pylate import evaluation, indexes, models, retrieve |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
"""Evaluation script for the SciFact dataset using the Beir library.""" | |
from __future__ import annotations | |
import os | |
import mteb | |
import srsly | |
from pylate import evaluation, indexes, models, retrieve |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from datasets import load_dataset | |
from sentence_transformers import ( | |
SentenceTransformerTrainer, | |
SentenceTransformerTrainingArguments, | |
) | |
from pylate import evaluation, losses, models, utils | |
# Load the datasets required for knowledge distillation (train, queries, documents) | |
train = load_dataset( |