lint

getnikola · May 22, 2017 · 0516994 · 0516994
1 parent e7fb38e
commit 0516994
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/v7/similarity/similarity.py b/v7/similarity/similarity.py
@@ -30,33 +30,32 @@
 
 from nikola.plugin_categories import Task
 
+
 class Similarity(Task):
     """Calculate post similarity."""
     name = "similarity"
 
     def set_site(self, site):
         self.site = site
-        
+
     def gen_tasks(self):
         """Build similarity data for each post."""
         self.site.scan_posts()
-        
+
         texts = []
-        
+
         for p in self.site.timeline:
             texts.append(p.text(strip_html=True).lower().split())
-        
+
         dictionary = gensim.corpora.Dictionary(texts)
         corpus = [dictionary.doc2bow(text) for text in texts]
         lsi = gensim.models.LsiModel(corpus, id2word=dictionary, num_topics=2)
         index = gensim.similarities.MatrixSimilarity(lsi[corpus])
-        
+
         for i, post in enumerate(self.site.timeline):
             doc = texts[i]
             vec_bow = dictionary.doc2bow(doc)
             vec_lsi = lsi[vec_bow]
             sims = index[vec_lsi]
             sims = sorted(enumerate(sims), key=lambda item: -item[1])
             print(i, sims[:10])
-
-