Add application file

tag_data/cal.py

@@ -2,7 +2,7 @@ import numpy as np
 from sentence_transformers import SentenceTransformer
 
 
-encoder = SentenceTransformer("hfl/chinese-roberta-wwm-ext-large")
+encoder = SentenceTransformer("./model/")
 
 tags = []
 f = open('tags.txt', 'r')
@@ -20,3 +20,16 @@ with open('./tags_embed.npy', 'wb') as f:
 
 with open('./tags_dis.npy', 'wb') as f:
     np.save(f, tags_dis)
+
+# f = open('gpttag.txt', 'r')
+# data = eval(f.readline())
+# f.close()
+#
+# out = ""
+# f = open('tags.txt', 'w')
+# for tmp in data:
+#     if tmp[1] > 2:
+#         out += tmp[0] + '\n'
+#
+# f.write(out)
+# f.close()