Ejemplos de DocumentSimilarity.nn en Python

Ejemplo n.º 1

Archivo: test_docsim.py Proyecto: 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)

Ejemplo n.º 2

Archivo: test_docsim.py Proyecto: 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)