本文整理汇总了Python中classifier.Classifier.load_trainSet方法的典型用法代码示例。如果您正苦于以下问题:Python Classifier.load_trainSet方法的具体用法?Python Classifier.load_trainSet怎么用?Python Classifier.load_trainSet使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类classifier.Classifier
的用法示例。
在下文中一共展示了Classifier.load_trainSet方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: create_predict
# 需要导入模块: from classifier import Classifier [as 别名]
# 或者: from classifier.Classifier import load_trainSet [as 别名]
def create_predict(HudongItem_csv):
# 读取neo4j内容
db = Neo4j()
db.connectDB()
data_set = db.getLabeledHudongItem('labels.txt')
classifier = Classifier('wiki.zh.bin')
classifier.load_trainSet(data_set)
classifier.set_parameter(weight=[1.0, 3.0, 0.2, 4.0, 0],k=10)
predict_List = readCSVbyColumn(HudongItem_csv, 'title')
file_object = open('predict_labels2.txt','a')
count = 0
vis = set()
for p in predict_List:
cur = HudongItem(db.matchHudongItembyTitle(p))
count += 1
title = cur.title
if title in vis:
continue
vis.add(title)
label = classifier.KNN_predict(cur)
print(str(title)+" "+str(label)+": "+str(count)+"/"+str(len(predict_List)))
file_object.write(str(title)+" "+str(label)+"\n")
file_object.close()
#create_predict('hudong_pedia2.csv')