|
@@ -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
|
|