Exemplos de DocumentSimilarity.nn em Python

Exemplo n.º 1

Arquivo: test_docsim.py Projeto: 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)

Exemplo n.º 2

Arquivo: test_docsim.py Projeto: 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)