@@ -620,11 +620,14 @@ The :class:`~sentence_transformers.cross_encoder.trainer.CrossEncoderTrainer` is
620
620
621
621
from datasets import load_dataset
622
622
623
- from sentence_transformers.cross_encoder import CrossEncoder, CrossEncoderModelCardData
623
+ from sentence_transformers.cross_encoder import (
624
+ CrossEncoder,
625
+ CrossEncoderModelCardData,
626
+ CrossEncoderTrainer,
627
+ CrossEncoderTrainingArguments,
628
+ )
624
629
from sentence_transformers.cross_encoder.evaluation import CrossEncoderNanoBEIREvaluator
625
630
from sentence_transformers.cross_encoder.losses import CachedMultipleNegativesRankingLoss
626
- from sentence_transformers.cross_encoder.trainer import CrossEncoderTrainer
627
- from sentence_transformers.cross_encoder.training_args import CrossEncoderTrainingArguments
628
631
629
632
# Set the log level to INFO to get more information
630
633
logging.basicConfig(format="%(asctime)s - %(message)s", datefmt="%Y-%m-%d %H:%M:%S", level=logging.INFO)
@@ -759,15 +762,18 @@ The :class:`~sentence_transformers.cross_encoder.trainer.CrossEncoderTrainer` is
759
762
from datasets import load_dataset
760
763
761
764
from sentence_transformers import SentenceTransformer
762
- from sentence_transformers.cross_encoder import CrossEncoder, CrossEncoderModelCardData
765
+ from sentence_transformers.cross_encoder import (
766
+ CrossEncoder,
767
+ CrossEncoderModelCardData,
768
+ CrossEncoderTrainer,
769
+ CrossEncoderTrainingArguments,
770
+ )
763
771
from sentence_transformers.cross_encoder.evaluation import (
764
772
CrossEncoderNanoBEIREvaluator,
765
773
CrossEncoderRerankingEvaluator,
766
774
)
767
- from sentence_transformers.cross_encoder.losses.BinaryCrossEntropyLoss import BinaryCrossEntropyLoss
768
- from sentence_transformers.cross_encoder.trainer import CrossEncoderTrainer
769
- from sentence_transformers.cross_encoder.training_args import CrossEncoderTrainingArguments
770
- from sentence_transformers.evaluation.SequentialEvaluator import SequentialEvaluator
775
+ from sentence_transformers.cross_encoder.losses import BinaryCrossEntropyLoss
776
+ from sentence_transformers.evaluation import SequentialEvaluator
771
777
from sentence_transformers.util import mine_hard_negatives
772
778
773
779
# Set the log level to INFO to get more information
0 commit comments