yufeng0528 4 vuotta sitten
vanhempi
commit
ee6ca69ff8
1 muutettua tiedostoa jossa 3 lisäystä ja 3 poistoa
  1. 3 3
      lda/save_annoy.py

+ 3 - 3
lda/save_annoy.py

@@ -6,7 +6,7 @@ length = 100
6 6
 t = AnnoyIndex(length,metric="angular")
7 7
 index_url={}
8 8
 with open("article_topics") as f:
9
-	lines=f.readlines()
9
+    lines=f.readlines()
10 10
 lines=[line.strip() for line in lines]
11 11
 for line in lines:
12 12
     data=json.loads(line)
@@ -19,7 +19,7 @@ for line in lines:
19 19
         t.add_item(aid, feature)
20 20
 t.build(10) 
21 21
 t.save('article.ann')
22
-with open("index_url","w") as f:
23
-    pickle.dump(index_url,f)
22
+with open("index_url","wb+") as f:
23
+    pickle.dump(url, f)
24 24
 
25 25