vijayarulmuthu's picture
Add new SentenceTransformer model
e1ca736 verified
metadata
tags:
  - sentence-transformers
  - sentence-similarity
  - feature-extraction
  - generated_from_trainer
  - dataset_size:6612
  - loss:MatryoshkaLoss
  - loss:MultipleNegativesRankingLoss
base_model: Snowflake/snowflake-arctic-embed-m
widget:
  - source_sentence: Who besought that the words might be preached to them the next sabbath?
    sentences:
      - >-
        But the midwives feared God, and did not as the king of Egypt commanded
        them, but saved the men children alive. And the king of Egypt called for
        the midwives, and said unto them, Why have ye done this thing, and have
        saved the men children alive? And the midwives said unto Pharaoh,
        Because the Hebrew women [are] not as the Egyptian women; for they [are]
        lively, and are delivered ere the midwives come in unto them. Therefore
        God dealt well with the midwives: and the people multiplied, and waxed
        very mighty. And it came to pass, because the midwives feared God, that
        he made them houses. And Pharaoh charged all his people, saying, Every
        son that is born ye shall cast into the river, and every daughter ye
        shall save alive.
      - >-
        And the watchman cried, and told the king. And the king said, If he [be]
        alone, [there is] tidings in his mouth. And he came apace, and drew
        near. And the watchman saw another man running: and the watchman called
        unto the porter, and said, Behold [another] man running alone. And the
        king said, He also bringeth tidings. And the watchman said, Me thinketh
        the running of the foremost is like the running of Ahimaaz the son of
        Zadok. And the king said, He [is] a good man, and cometh with good
        tidings. And Ahimaaz called, and said unto the king, All is well. And he
        fell down to the earth upon his face before the king, and said, Blessed
        [be] the LORD thy God, which hath delivered up the men that lifted up
        their hand against my lord the king. And the king said, [Is] the young
        man Absalom safe? And Ahimaaz answered, When Joab sent the king’s
        servant, and [me] thy servant, I saw a great tumult, but I knew not what
        [it was]. And the king said [unto him], Turn aside, [and] stand here.
        And he turned aside, and stood still. And, behold, Cushi came; and Cushi
        said, Tidings, my lord the king: for the LORD hath avenged thee this day
        of all them that rose up against thee. And the king said unto Cushi,
        [Is] the young man Absalom safe? And Cushi answered, The enemies of my
        lord the king, and all that rise against thee to do [thee] hurt, be as
        [that] young man [is].
      - >-
        Behold, ye despisers, and wonder, and perish: for I work a work in your
        days, a work which ye shall in no wise believe, though a man declare it
        unto you. And when the Jews were gone out of the synagogue, the Gentiles
        besought that these words might be preached to them the next sabbath.
        Now when the congregation was broken up, many of the Jews and religious
        proselytes followed Paul and Barnabas: who, speaking to them, persuaded
        them to continue in the grace of God. And the next sabbath day came
        almost the whole city together to hear the word of God. But when the
        Jews saw the multitudes, they were filled with envy, and spake against
        those things which were spoken by Paul, contradicting and blaspheming.
        Then Paul and Barnabas waxed bold, and said, It was necessary that the
        word of God should first have been spoken to you: but seeing ye put it
        from you, and judge yourselves unworthy of everlasting life, lo, we turn
        to the Gentiles. For so hath the Lord commanded us, [saying], I have set
        thee to be a light of the Gentiles, that thou shouldest be for salvation
        unto the ends of the earth. And when the Gentiles heard this, they were
        glad, and glorified the word of the Lord: and as many as were ordained
        to eternal life believed.
  - source_sentence: >-
      What will the LORD do if the people hearken unto Him and keep His
      commandments?
    sentences:
      - >-
        And ye shall not go out of the door of the tabernacle of the
        congregation [in] seven days, until the days of your consecration be at
        an end: for seven days shall he consecrate you. As he hath done this
        day, [so] the LORD hath commanded to do, to make an atonement for you.
        Therefore shall ye abide [at] the door of the tabernacle of the
        congregation day and night seven days, and keep the charge of the LORD,
        that ye die not: for so I am commanded. So Aaron and his sons did all
        things which the LORD commanded by the hand of Moses.
      - >-
        Then all Israel gathered themselves to David unto Hebron, saying,
        Behold, we [are] thy bone and thy flesh. And moreover in time past, even
        when Saul was king, thou [wast] he that leddest out and broughtest in
        Israel: and the LORD thy God said unto thee, Thou shalt feed my people
        Israel, and thou shalt be ruler over my people Israel. Therefore came
        all the elders of Israel to the king to Hebron; and David made a
        covenant with them in Hebron before the LORD; and they anointed David
        king over Israel, according to the word of the LORD by Samuel. And David
        and all Israel went to Jerusalem, which [is] Jebus; where the Jebusites
        [were], the inhabitants of the land. And the inhabitants of Jebus said
        to David, Thou shalt not come hither. Nevertheless David took the castle
        of Zion, which [is] the city of David. And David said, Whosoever smiteth
        the Jebusites first shall be chief and captain. So Joab the son of
        Zeruiah went first up, and was chief. And David dwelt in the castle;
        therefore they called it the city of David. And he built the city round
        about, even from Millo round about: and Joab repaired the rest of the
        city.
      - >-
        For I will have respect unto you, and make you fruitful, and multiply
        you, and establish my covenant with you. And ye shall eat old store, and
        bring forth the old because of the new. And I will set my tabernacle
        among you: and my soul shall not abhor you. And I will walk among you,
        and will be your God, and ye shall be my people. I [am] the LORD your
        God, which brought you forth out of the land of Egypt, that ye should
        not be their bondmen; and I have broken the bands of your yoke, and made
        you go upright. But if ye will not hearken unto me, and will not do all
        these commandments; And if ye shall despise my statutes, or if your soul
        abhor my judgments, so that ye will not do all my commandments, [but]
        that ye break my covenant: I also will do this unto you; I will even
        appoint over you terror, consumption, and the burning ague, that shall
        consume the eyes, and cause sorrow of heart: and ye shall sow your seed
        in vain, for your enemies shall eat it.
  - source_sentence: How are the nations that fight against Ariel described in their outcome?
    sentences:
      - >-
        Then the LORD put forth his hand, and touched my mouth. And the LORD
        said unto me, Behold, I have put my words in thy mouth. See, I have this
        day set thee over the nations and over the kingdoms, to root out, and to
        pull down, and to destroy, and to throw down, to build, and to plant.
        Moreover the word of the LORD came unto me, saying, Jeremiah, what seest
        thou? And I said, I see a rod of an almond tree. Then said the LORD unto
        me, Thou hast well seen: for I will hasten my word to perform it. And
        the word of the LORD came unto me the second time, saying, What seest
        thou? And I said, I see a seething pot; and the face thereof [is] toward
        the north. Then the LORD said unto me, Out of the north an evil shall
        break forth upon all the inhabitants of the land. For, lo, I will call
        all the families of the kingdoms of the north, saith the LORD; and they
        shall come, and they shall set every one his throne at the entering of
        the gates of Jerusalem, and against all the walls thereof round about,
        and against all the cities of Judah. And I will utter my judgments
        against them touching all their wickedness, who have forsaken me, and
        have burned incense unto other gods, and worshipped the works of their
        own hands.
      - >-
        Woe to Ariel, to Ariel, the city [where] David dwelt! add ye year to
        year; let them kill sacrifices. Yet I will distress Ariel, and there
        shall be heaviness and sorrow: and it shall be unto me as Ariel. And I
        will camp against thee round about, and will lay siege against thee with
        a mount, and I will raise forts against thee. And thou shalt be brought
        down, [and] shalt speak out of the ground, and thy speech shall be low
        out of the dust, and thy voice shall be, as of one that hath a familiar
        spirit, out of the ground, and thy speech shall whisper out of the dust.
        Moreover the multitude of thy strangers shall be like small dust, and
        the multitude of the terrible ones [shall be] as chaff that passeth
        away: yea, it shall be at an instant suddenly. Thou shalt be visited of
        the LORD of hosts with thunder, and with earthquake, and great noise,
        with storm and tempest, and the flame of devouring fire. And the
        multitude of all the nations that fight against Ariel, even all that
        fight against her and her munition, and that distress her, shall be as a
        dream of a night vision. It shall even be as when an hungry [man]
        dreameth, and, behold, he eateth; but he awaketh, and his soul is empty:
        or as when a thirsty man dreameth, and, behold, he drinketh; but he
        awaketh, and, behold, [he is] faint, and his soul hath appetite: so
        shall the multitude of all the nations be, that fight against mount
        Zion.
      - And they were both naked, the man and his wife, and were not ashamed.
  - source_sentence: What will happen to the horn and arm of Moab according to the LORD?
    sentences:
      - >-
        The horn of Moab is cut off, and his arm is broken, saith the LORD. Make
        ye him drunken: for he magnified [himself] against the LORD: Moab also
        shall wallow in his vomit, and he also shall be in derision. For was not
        Israel a derision unto thee? was he found among thieves? for since thou
        spakest of him, thou skippedst for joy. O ye that dwell in Moab, leave
        the cities, and dwell in the rock, and be like the dove [that] maketh
        her nest in the sides of the hole’s mouth. We have heard the pride of
        Moab, (he is exceeding proud) his loftiness, and his arrogancy, and his
        pride, and the haughtiness of his heart. I know his wrath, saith the
        LORD; but [it shall] not [be] so; his lies shall not so effect [it].
        Therefore will I howl for Moab, and I will cry out for all Moab; [mine
        heart] shall mourn for the men of Kirheres. O vine of Sibmah, I will
        weep for thee with the weeping of Jazer: thy plants are gone over the
        sea, they reach [even] to the sea of Jazer: the spoiler is fallen upon
        thy summer fruits and upon thy vintage.
      - >-
        Wherefore be ye not unwise, but understanding what the will of the Lord
        [is]. And be not drunk with wine, wherein is excess; but be filled with
        the Spirit; Speaking to yourselves in psalms and hymns and spiritual
        songs, singing and making melody in your heart to the Lord; Giving
        thanks always for all things unto God and the Father in the name of our
        Lord Jesus Christ; Submitting yourselves one to another in the fear of
        God. Wives, submit yourselves unto your own husbands, as unto the Lord.
        For the husband is the head of the wife, even as Christ is the head of
        the church: and he is the saviour of the body. Therefore as the church
        is subject unto Christ, so [let] the wives [be] to their own husbands in
        every thing.
      - >-
        And it came to pass, when the LORD would take up Elijah into heaven by a
        whirlwind, that Elijah went with Elisha from Gilgal. And Elijah said
        unto Elisha, Tarry here, I pray thee; for the LORD hath sent me to
        Bethel. And Elisha said [unto him, As] the LORD liveth, and [as] thy
        soul liveth, I will not leave thee. So they went down to Bethel. And the
        sons of the prophets that [were] at Bethel came forth to Elisha, and
        said unto him, Knowest thou that the LORD will take away thy master from
        thy head to day? And he said, Yea, I know [it]; hold ye your peace. And
        Elijah said unto him, Elisha, tarry here, I pray thee; for the LORD hath
        sent me to Jericho. And he said, [As] the LORD liveth, and [as] thy soul
        liveth, I will not leave thee. So they came to Jericho. And the sons of
        the prophets that [were] at Jericho came to Elisha, and said unto him,
        Knowest thou that the LORD will take away thy master from thy head to
        day? And he answered, Yea, I know [it]; hold ye your peace. And Elijah
        said unto him, Tarry, I pray thee, here; for the LORD hath sent me to
        Jordan. And he said, [As] the LORD liveth, and [as] thy soul liveth, I
        will not leave thee. And they two went on. And fifty men of the sons of
        the prophets went, and stood to view afar off: and they two stood by
        Jordan. And Elijah took his mantle, and wrapped [it] together, and smote
        the waters, and they were divided hither and thither, so that they two
        went over on dry ground.
  - source_sentence: >-
      Whom did David smite and subdue, taking Gath and her towns from their
      control?
    sentences:
      - >-
        Remember, I beseech thee, that thou hast made me as the clay; and wilt
        thou bring me into dust again? Hast thou not poured me out as milk, and
        curdled me like cheese? Thou hast clothed me with skin and flesh, and
        hast fenced me with bones and sinews. Thou hast granted me life and
        favour, and thy visitation hath preserved my spirit. And these [things]
        hast thou hid in thine heart: I know that this [is] with thee. If I sin,
        then thou markest me, and thou wilt not acquit me from mine iniquity. If
        I be wicked, woe unto me; and [if] I be righteous, [yet] will I not lift
        up my head. [I am] full of confusion; therefore see thou mine
        affliction; For it increaseth. Thou huntest me as a fierce lion: and
        again thou shewest thyself marvellous upon me.
      - >-
        Now after this it came to pass, that David smote the Philistines, and
        subdued them, and took Gath and her towns out of the hand of the
        Philistines. And he smote Moab; and the Moabites became David’s
        servants, [and] brought gifts. And David smote Hadarezer king of Zobah
        unto Hamath, as he went to stablish his dominion by the river Euphrates.
        And David took from him a thousand chariots, and seven thousand
        horsemen, and twenty thousand footmen: David also houghed all the
        chariot [horses], but reserved of them an hundred chariots. And when the
        Syrians of Damascus came to help Hadarezer king of Zobah, David slew of
        the Syrians two and twenty thousand men. Then David put [garrisons] in
        Syriadamascus; and the Syrians became David’s servants, [and] brought
        gifts. Thus the LORD preserved David whithersoever he went. And David
        took the shields of gold that were on the servants of Hadarezer, and
        brought them to Jerusalem. Likewise from Tibhath, and from Chun, cities
        of Hadarezer, brought David very much brass, wherewith Solomon made the
        brasen sea, and the pillars, and the vessels of brass.
      - >-
        So Shishak king of Egypt came up against Jerusalem, and took away the
        treasures of the house of the LORD, and the treasures of the king’s
        house; he took all: he carried away also the shields of gold which
        Solomon had made. Instead of which king Rehoboam made shields of brass,
        and committed [them] to the hands of the chief of the guard, that kept
        the entrance of the king’s house. And when the king entered into the
        house of the LORD, the guard came and fetched them, and brought them
        again into the guard chamber. And when he humbled himself, the wrath of
        the LORD turned from him, that he would not destroy [him] altogether:
        and also in Judah things went well. So king Rehoboam strengthened
        himself in Jerusalem, and reigned: for Rehoboam [was] one and forty
        years old when he began to reign, and he reigned seventeen years in
        Jerusalem, the city which the LORD had chosen out of all the tribes of
        Israel, to put his name there. And his mother’s name [was] Naamah an
        Ammonitess. And he did evil, because he prepared not his heart to seek
        the LORD. Now the acts of Rehoboam, first and last, [are] they not
        written in the book of Shemaiah the prophet, and of Iddo the seer
        concerning genealogies? And [there were] wars between Rehoboam and
        Jeroboam continually. And Rehoboam slept with his fathers, and was
        buried in the city of David: and Abijah his son reigned in his stead.
pipeline_tag: sentence-similarity
library_name: sentence-transformers
metrics:
  - cosine_accuracy@1
  - cosine_accuracy@3
  - cosine_accuracy@5
  - cosine_accuracy@10
  - cosine_precision@1
  - cosine_precision@3
  - cosine_precision@5
  - cosine_precision@10
  - cosine_recall@1
  - cosine_recall@3
  - cosine_recall@5
  - cosine_recall@10
  - cosine_ndcg@10
  - cosine_mrr@10
  - cosine_map@100
model-index:
  - name: SentenceTransformer based on Snowflake/snowflake-arctic-embed-m
    results:
      - task:
          type: information-retrieval
          name: Information Retrieval
        dataset:
          name: validation
          type: validation
        metrics:
          - type: cosine_accuracy@1
            value: 0.647912885662432
            name: Cosine Accuracy@1
          - type: cosine_accuracy@3
            value: 0.8330308529945554
            name: Cosine Accuracy@3
          - type: cosine_accuracy@5
            value: 0.8729582577132486
            name: Cosine Accuracy@5
          - type: cosine_accuracy@10
            value: 0.9237749546279492
            name: Cosine Accuracy@10
          - type: cosine_precision@1
            value: 0.647912885662432
            name: Cosine Precision@1
          - type: cosine_precision@3
            value: 0.2776769509981851
            name: Cosine Precision@3
          - type: cosine_precision@5
            value: 0.17459165154264972
            name: Cosine Precision@5
          - type: cosine_precision@10
            value: 0.0923774954627949
            name: Cosine Precision@10
          - type: cosine_recall@1
            value: 0.017997580157289778
            name: Cosine Recall@1
          - type: cosine_recall@3
            value: 0.02313974591651543
            name: Cosine Recall@3
          - type: cosine_recall@5
            value: 0.024248840492034688
            name: Cosine Recall@5
          - type: cosine_recall@10
            value: 0.02566041540633193
            name: Cosine Recall@10
          - type: cosine_ndcg@10
            value: 0.17391225053060402
            name: Cosine Ndcg@10
          - type: cosine_mrr@10
            value: 0.7468556448592748
            name: Cosine Mrr@10
          - type: cosine_map@100
            value: 0.020838529961750816
            name: Cosine Map@100

SentenceTransformer based on Snowflake/snowflake-arctic-embed-m

This is a sentence-transformers model finetuned from Snowflake/snowflake-arctic-embed-m. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.

Model Details

Model Description

  • Model Type: Sentence Transformer
  • Base model: Snowflake/snowflake-arctic-embed-m
  • Maximum Sequence Length: 512 tokens
  • Output Dimensionality: 768 dimensions
  • Similarity Function: Cosine Similarity

Model Sources

Full Model Architecture

SentenceTransformer(
  (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: BertModel 
  (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': True, 'pooling_mode_mean_tokens': False, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
  (2): Normalize()
)

Usage

Direct Usage (Sentence Transformers)

First install the Sentence Transformers library:

pip install -U sentence-transformers

Then you can load this model and run inference.

from sentence_transformers import SentenceTransformer

# Download from the 🤗 Hub
model = SentenceTransformer("vijayarulmuthu/finetuned_arctic_ft-a85433c9-6284-4afb-8e87-e110823d565c")
# Run inference
sentences = [
    'Whom did David smite and subdue, taking Gath and her towns from their control?',
    'Now after this it came to pass, that David smote the Philistines, and subdued them, and took Gath and her towns out of the hand of the Philistines. And he smote Moab; and the Moabites became David’s servants, [and] brought gifts. And David smote Hadarezer king of Zobah unto Hamath, as he went to stablish his dominion by the river Euphrates. And David took from him a thousand chariots, and seven thousand horsemen, and twenty thousand footmen: David also houghed all the chariot [horses], but reserved of them an hundred chariots. And when the Syrians of Damascus came to help Hadarezer king of Zobah, David slew of the Syrians two and twenty thousand men. Then David put [garrisons] in Syriadamascus; and the Syrians became David’s servants, [and] brought gifts. Thus the LORD preserved David whithersoever he went. And David took the shields of gold that were on the servants of Hadarezer, and brought them to Jerusalem. Likewise from Tibhath, and from Chun, cities of Hadarezer, brought David very much brass, wherewith Solomon made the brasen sea, and the pillars, and the vessels of brass.',
    'So Shishak king of Egypt came up against Jerusalem, and took away the treasures of the house of the LORD, and the treasures of the king’s house; he took all: he carried away also the shields of gold which Solomon had made. Instead of which king Rehoboam made shields of brass, and committed [them] to the hands of the chief of the guard, that kept the entrance of the king’s house. And when the king entered into the house of the LORD, the guard came and fetched them, and brought them again into the guard chamber. And when he humbled himself, the wrath of the LORD turned from him, that he would not destroy [him] altogether: and also in Judah things went well. So king Rehoboam strengthened himself in Jerusalem, and reigned: for Rehoboam [was] one and forty years old when he began to reign, and he reigned seventeen years in Jerusalem, the city which the LORD had chosen out of all the tribes of Israel, to put his name there. And his mother’s name [was] Naamah an Ammonitess. And he did evil, because he prepared not his heart to seek the LORD. Now the acts of Rehoboam, first and last, [are] they not written in the book of Shemaiah the prophet, and of Iddo the seer concerning genealogies? And [there were] wars between Rehoboam and Jeroboam continually. And Rehoboam slept with his fathers, and was buried in the city of David: and Abijah his son reigned in his stead.',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]

# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]

Evaluation

Metrics

Information Retrieval

Metric Value
cosine_accuracy@1 0.6479
cosine_accuracy@3 0.833
cosine_accuracy@5 0.873
cosine_accuracy@10 0.9238
cosine_precision@1 0.6479
cosine_precision@3 0.2777
cosine_precision@5 0.1746
cosine_precision@10 0.0924
cosine_recall@1 0.018
cosine_recall@3 0.0231
cosine_recall@5 0.0242
cosine_recall@10 0.0257
cosine_ndcg@10 0.1739
cosine_mrr@10 0.7469
cosine_map@100 0.0208

Training Details

Training Dataset

Unnamed Dataset

  • Size: 6,612 training samples
  • Columns: sentence_0 and sentence_1
  • Approximate statistics based on the first 1000 samples:
    sentence_0 sentence_1
    type string string
    details
    • min: 7 tokens
    • mean: 17.56 tokens
    • max: 42 tokens
    • min: 13 tokens
    • mean: 250.95 tokens
    • max: 504 tokens
  • Samples:
    sentence_0 sentence_1
    What was the reason given by Elijah the prophet for the LORD's punishment on Jehoram? Then Jehoram went forth with his princes, and all his chariots with him: and he rose up by night, and smote the Edomites which compassed him in, and the captains of the chariots. So the Edomites revolted from under the hand of Judah unto this day. The same time [also] did Libnah revolt from under his hand; because he had forsaken the LORD God of his fathers. Moreover he made high places in the mountains of Judah, and caused the inhabitants of Jerusalem to commit fornication, and compelled Judah [thereto]. And there came a writing to him from Elijah the prophet, saying, Thus saith the LORD God of David thy father, Because thou hast not walked in the ways of Jehoshaphat thy father, nor in the ways of Asa king of Judah, But hast walked in the way of the kings of Israel, and hast made Judah and the inhabitants of Jerusalem to go a whoring, like to the whoredoms of the house of Ahab, and also hast slain thy brethren of thy father’s house, [which were] better than thyself: Behold, with a gre...
    What happened at the sixth hour until the ninth hour according to the passage? And we indeed justly; for we receive the due reward of our deeds: but this man hath done nothing amiss. And he said unto Jesus, Lord, remember me when thou comest into thy kingdom. And Jesus said unto him, Verily I say unto thee, To day shalt thou be with me in paradise. And it was about the sixth hour, and there was a darkness over all the earth until the ninth hour. And the sun was darkened, and the veil of the temple was rent in the midst. And when Jesus had cried with a loud voice, he said, Father, into thy hands I commend my spirit: and having said thus, he gave up the ghost. Now when the centurion saw what was done, he glorified God, saying, Certainly this was a righteous man. And all the people that came together to that sight, beholding the things which were done, smote their breasts, and returned.
    Who is commanded by the Lord to set a watchman and declare what he sees? The burden of the desert of the sea. As whirlwinds in the south pass through; [so] it cometh from the desert, from a terrible land. A grievous vision is declared unto me; the treacherous dealer dealeth treacherously, and the spoiler spoileth. Go up, O Elam: besiege, O Media; all the sighing thereof have I made to cease. Therefore are my loins filled with pain: pangs have taken hold upon me, as the pangs of a woman that travaileth: I was bowed down at the hearing [of it]; I was dismayed at the seeing [of it]. My heart panted, fearfulness affrighted me: the night of my pleasure hath he turned into fear unto me. Prepare the table, watch in the watchtower, eat, drink: arise, ye princes, [and] anoint the shield. For thus hath the Lord said unto me, Go, set a watchman, let him declare what he seeth. And he saw a chariot [with] a couple of horsemen, a chariot of asses, [and] a chariot of camels; and he hearkened diligently with much heed: And he cried, A lion: My lord, I stand continually upo...
  • Loss: MatryoshkaLoss with these parameters:
    {
        "loss": "MultipleNegativesRankingLoss",
        "matryoshka_dims": [
            768,
            512,
            256,
            128,
            64
        ],
        "matryoshka_weights": [
            1,
            1,
            1,
            1,
            1
        ],
        "n_dims_per_step": -1
    }
    

Training Hyperparameters

Non-Default Hyperparameters

  • eval_strategy: steps
  • per_device_train_batch_size: 10
  • per_device_eval_batch_size: 10
  • num_train_epochs: 10
  • multi_dataset_batch_sampler: round_robin

All Hyperparameters

Click to expand
  • overwrite_output_dir: False
  • do_predict: False
  • eval_strategy: steps
  • prediction_loss_only: True
  • per_device_train_batch_size: 10
  • per_device_eval_batch_size: 10
  • per_gpu_train_batch_size: None
  • per_gpu_eval_batch_size: None
  • gradient_accumulation_steps: 1
  • eval_accumulation_steps: None
  • torch_empty_cache_steps: None
  • learning_rate: 5e-05
  • weight_decay: 0.0
  • adam_beta1: 0.9
  • adam_beta2: 0.999
  • adam_epsilon: 1e-08
  • max_grad_norm: 1
  • num_train_epochs: 10
  • max_steps: -1
  • lr_scheduler_type: linear
  • lr_scheduler_kwargs: {}
  • warmup_ratio: 0.0
  • warmup_steps: 0
  • log_level: passive
  • log_level_replica: warning
  • log_on_each_node: True
  • logging_nan_inf_filter: True
  • save_safetensors: True
  • save_on_each_node: False
  • save_only_model: False
  • restore_callback_states_from_checkpoint: False
  • no_cuda: False
  • use_cpu: False
  • use_mps_device: False
  • seed: 42
  • data_seed: None
  • jit_mode_eval: False
  • use_ipex: False
  • bf16: False
  • fp16: False
  • fp16_opt_level: O1
  • half_precision_backend: auto
  • bf16_full_eval: False
  • fp16_full_eval: False
  • tf32: None
  • local_rank: 0
  • ddp_backend: None
  • tpu_num_cores: None
  • tpu_metrics_debug: False
  • debug: []
  • dataloader_drop_last: False
  • dataloader_num_workers: 0
  • dataloader_prefetch_factor: None
  • past_index: -1
  • disable_tqdm: False
  • remove_unused_columns: True
  • label_names: None
  • load_best_model_at_end: False
  • ignore_data_skip: False
  • fsdp: []
  • fsdp_min_num_params: 0
  • fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
  • fsdp_transformer_layer_cls_to_wrap: None
  • accelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
  • deepspeed: None
  • label_smoothing_factor: 0.0
  • optim: adamw_torch
  • optim_args: None
  • adafactor: False
  • group_by_length: False
  • length_column_name: length
  • ddp_find_unused_parameters: None
  • ddp_bucket_cap_mb: None
  • ddp_broadcast_buffers: False
  • dataloader_pin_memory: True
  • dataloader_persistent_workers: False
  • skip_memory_metrics: True
  • use_legacy_prediction_loop: False
  • push_to_hub: False
  • resume_from_checkpoint: None
  • hub_model_id: None
  • hub_strategy: every_save
  • hub_private_repo: None
  • hub_always_push: False
  • gradient_checkpointing: False
  • gradient_checkpointing_kwargs: None
  • include_inputs_for_metrics: False
  • include_for_metrics: []
  • eval_do_concat_batches: True
  • fp16_backend: auto
  • push_to_hub_model_id: None
  • push_to_hub_organization: None
  • mp_parameters:
  • auto_find_batch_size: False
  • full_determinism: False
  • torchdynamo: None
  • ray_scope: last
  • ddp_timeout: 1800
  • torch_compile: False
  • torch_compile_backend: None
  • torch_compile_mode: None
  • include_tokens_per_second: False
  • include_num_input_tokens_seen: False
  • neftune_noise_alpha: None
  • optim_target_modules: None
  • batch_eval_metrics: False
  • eval_on_start: False
  • use_liger_kernel: False
  • eval_use_gather_object: False
  • average_tokens_across_devices: False
  • prompts: None
  • batch_sampler: batch_sampler
  • multi_dataset_batch_sampler: round_robin

Training Logs

Click to expand
Epoch Step Training Loss validation_cosine_ndcg@10
0.0755 50 - 0.0982
0.1511 100 - 0.1408
0.2266 150 - 0.1546
0.3021 200 - 0.1612
0.3776 250 - 0.1655
0.4532 300 - 0.1663
0.5287 350 - 0.1710
0.6042 400 - 0.1704
0.6798 450 - 0.1713
0.7553 500 2.378 0.1702
0.8308 550 - 0.1727
0.9063 600 - 0.1734
0.9819 650 - 0.1741
1.0 662 - 0.1745
1.0574 700 - 0.1752
1.1329 750 - 0.1761
1.2085 800 - 0.1750
1.2840 850 - 0.1719
1.3595 900 - 0.1730
1.4350 950 - 0.1760
1.5106 1000 0.7402 0.1776
1.5861 1050 - 0.1757
1.6616 1100 - 0.1774
1.7372 1150 - 0.1757
1.8127 1200 - 0.1749
1.8882 1250 - 0.1745
1.9637 1300 - 0.1758
2.0 1324 - 0.1776
2.0393 1350 - 0.1772
2.1148 1400 - 0.1751
2.1903 1450 - 0.1757
2.2659 1500 0.467 0.1742
2.3414 1550 - 0.1748
2.4169 1600 - 0.1738
2.4924 1650 - 0.1749
2.5680 1700 - 0.1772
2.6435 1750 - 0.1772
2.7190 1800 - 0.1772
2.7946 1850 - 0.1774
2.8701 1900 - 0.1770
2.9456 1950 - 0.1757
3.0 1986 - 0.1771
3.0211 2000 0.2653 0.1762
3.0967 2050 - 0.1745
3.1722 2100 - 0.1748
3.2477 2150 - 0.1749
3.3233 2200 - 0.1766
3.3988 2250 - 0.1746
3.4743 2300 - 0.1749
3.5498 2350 - 0.1766
3.6254 2400 - 0.1752
3.7009 2450 - 0.1749
3.7764 2500 0.1809 0.1746
3.8520 2550 - 0.1751
3.9275 2600 - 0.1755
4.0 2648 - 0.1744
4.0030 2650 - 0.1747
4.0785 2700 - 0.1747
4.1541 2750 - 0.1766
4.2296 2800 - 0.1761
4.3051 2850 - 0.1745
4.3807 2900 - 0.1748
4.4562 2950 - 0.1753
4.5317 3000 0.1368 0.1741
4.6073 3050 - 0.1718
4.6828 3100 - 0.1730
4.7583 3150 - 0.1735
4.8338 3200 - 0.1753
4.9094 3250 - 0.1744
4.9849 3300 - 0.1752
5.0 3310 - 0.1758
5.0604 3350 - 0.1771
5.1360 3400 - 0.1758
5.2115 3450 - 0.1741
5.2870 3500 0.1178 0.1741
5.3625 3550 - 0.1746
5.4381 3600 - 0.1744
5.5136 3650 - 0.1740
5.5891 3700 - 0.1743
5.6647 3750 - 0.1744
5.7402 3800 - 0.1733
5.8157 3850 - 0.1747
5.8912 3900 - 0.1755
5.9668 3950 - 0.1734
6.0 3972 - 0.1740
6.0423 4000 0.0878 0.1745
6.1178 4050 - 0.1734
6.1934 4100 - 0.1725
6.2689 4150 - 0.1748
6.3444 4200 - 0.1743
6.4199 4250 - 0.1742
6.4955 4300 - 0.1738
6.5710 4350 - 0.1756
6.6465 4400 - 0.1746
6.7221 4450 - 0.1754
6.7976 4500 0.0697 0.1756
6.8731 4550 - 0.1755
6.9486 4600 - 0.1755
7.0 4634 - 0.1755
7.0242 4650 - 0.1752
7.0997 4700 - 0.1766
7.1752 4750 - 0.1745
7.2508 4800 - 0.1751
7.3263 4850 - 0.1746
7.4018 4900 - 0.1747
7.4773 4950 - 0.1742
7.5529 5000 0.0643 0.1743
7.6284 5050 - 0.1736
7.7039 5100 - 0.1739
7.7795 5150 - 0.1737
7.8550 5200 - 0.1736
7.9305 5250 - 0.1744
8.0 5296 - 0.1750
8.0060 5300 - 0.1751
8.0816 5350 - 0.1742
8.1571 5400 - 0.1739
8.2326 5450 - 0.1745
8.3082 5500 0.0521 0.1745
8.3837 5550 - 0.1746
8.4592 5600 - 0.1743
8.5347 5650 - 0.1744
8.6103 5700 - 0.1750
8.6858 5750 - 0.1749
8.7613 5800 - 0.1748
8.8369 5850 - 0.1747
8.9124 5900 - 0.1747
8.9879 5950 - 0.1746
9.0 5958 - 0.1746
9.0634 6000 0.044 0.1745
9.1390 6050 - 0.1742
9.2145 6100 - 0.1740
9.2900 6150 - 0.1742
9.3656 6200 - 0.1744
9.4411 6250 - 0.1739
9.5166 6300 - 0.1737
9.5921 6350 - 0.1740
9.6677 6400 - 0.1738
9.7432 6450 - 0.1739
9.8187 6500 0.043 0.1738
9.8943 6550 - 0.1738
9.9698 6600 - 0.1739
10.0 6620 - 0.1739

Framework Versions

  • Python: 3.13.3
  • Sentence Transformers: 4.1.0
  • Transformers: 4.52.3
  • PyTorch: 2.7.0
  • Accelerate: 1.7.0
  • Datasets: 3.6.0
  • Tokenizers: 0.21.1

Citation

BibTeX

Sentence Transformers

@inproceedings{reimers-2019-sentence-bert,
    title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
    author = "Reimers, Nils and Gurevych, Iryna",
    booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
    month = "11",
    year = "2019",
    publisher = "Association for Computational Linguistics",
    url = "https://arxiv.org/abs/1908.10084",
}

MatryoshkaLoss

@misc{kusupati2024matryoshka,
    title={Matryoshka Representation Learning},
    author={Aditya Kusupati and Gantavya Bhatt and Aniket Rege and Matthew Wallingford and Aditya Sinha and Vivek Ramanujan and William Howard-Snyder and Kaifeng Chen and Sham Kakade and Prateek Jain and Ali Farhadi},
    year={2024},
    eprint={2205.13147},
    archivePrefix={arXiv},
    primaryClass={cs.LG}
}

MultipleNegativesRankingLoss

@misc{henderson2017efficient,
    title={Efficient Natural Language Response Suggestion for Smart Reply},
    author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
    year={2017},
    eprint={1705.00652},
    archivePrefix={arXiv},
    primaryClass={cs.CL}
}