similar-news-sentence-transformer1

Sleeping

ksvmuralidhar commited on Jan 13

Commit

f2f1bf9

•

1 Parent(s): d35aa2a

Update insert_into_db_sent_tran.ipynb

Files changed (1) hide show

insert_into_db_sent_tran.ipynb CHANGED Viewed

@@ -234,7 +234,7 @@
     "    sentence transformers to extract sentence embeddings\n",
     "    '''\n",
     "    def vectorize(self, x):\n",
-    "        sent_model = SentenceTransformer('multi-qa-distilbert-cos-v1')\n",
     "        sen_embeddings = sent_model.encode(x)\n",
     "        return sen_embeddings"
    ]

     "    sentence transformers to extract sentence embeddings\n",
     "    '''\n",
     "    def vectorize(self, x):\n",
+    "        sent_model = SentenceTransformer('all-mpnet-base-v2')\n",
     "        sen_embeddings = sent_model.encode(x)\n",
     "        return sen_embeddings"
    ]