mldr/models/retrieva.py

PROMPT = {
    "STS": "同じ意味の文を探すために次の文を表現して\n",
    "Summarization": "次の記事またはタイトルを表現して\n",
    "BitextMining": "次の文を表現して\n",
    "Classification": "同じクラスに属する文を探すために次の文を表現して\n",
    "Clustering": "類似した文を探すために次の文を表現して\n",
    "Reranking-query": "関連した文書を探すために次の文を表現して\n",
    "Reranking-passage": "次の文章を表現して\n",
    "Retrieval-query": "関連した文書を探すために次の文を表現して\n",
    "Retrieval-passage": "次の文章を表現して\n",
    "InstructionRetrieval": "",
    "PairClassification": "同じ意味の文を探すために次の文を表現して\n",
}