Python DocumentSimilarity.nn Beispiele

Beispiel #1

Datei: test_docsim.py Projekt: zofuthan/seldon-server

 def test_gensim_rp(self):
     corpus = DefaultJsonCorpus(self.get_docs())
     ds = DocumentSimilarity(model_type="gensim_rp")
     ds.fit(corpus)
     res = ds.nn(0,k=1)
     self.assertEqual(res[0][0],1)

Beispiel #2

Datei: test_docsim.py Projekt: zofuthan/seldon-server

 def test_sklearn_nmf(self):
     corpus = DefaultJsonCorpus(self.get_docs())
     ds = DocumentSimilarity(model_type="sklearn_nmf")
     ds.fit(corpus)
     res = ds.nn(0,k=1)
     self.assertEqual(res[0][0],1)