本文整理汇总了Python中classifier.Classifier.preprocess方法的典型用法代码示例。如果您正苦于以下问题:Python Classifier.preprocess方法的具体用法?Python Classifier.preprocess怎么用?Python Classifier.preprocess使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类classifier.Classifier
的用法示例。
在下文中一共展示了Classifier.preprocess方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: main
# 需要导入模块: from classifier import Classifier [as 别名]
# 或者: from classifier.Classifier import preprocess [as 别名]
def main():
print("started at %s"%str(time.localtime()))
if dumped in os.listdir():
with open(dumped,"rb") as f:
clf = pickle.load(f)
else:
clf = Classifier()
dataNameList = os.listdir(dataPath)
trainModels = [Model(name.split(".")[0], True) for name in dataNameList]
data = np.concatenate([m.feature for m in trainModels],axis=0)
label = np.concatenate([m.labels for m in trainModels],axis=0)
data = clf.preprocess(data, True)
clf.fit(data, label)
print("training time end at %s"%str(time.localtime()))
clf.save()
testNameList = os.listdir(testPath)
testModels = [Model(name.split(".")[0]) for name in testNameList]
for m in testModels:
data = m.feature
data = clf.preprocess(data)
print("compute probability.")
proba = clf.predict_proba(data)
print("saving prob")
np.savetxt("./result_prob/%s.prob"%m.name,proba)
print("saving mid-res.")
np.savetxt("./result_mid/%s.seg"%m.name,np.argmax(proba,axis=-1),fmt="%d")
print("cutting")
WG = build_weight_graph(m.faceGraph,clf.transform(data, threshold="median"))
gList = build_s_t_graph_list(WG, proba)
result = cut_and_label(gList)
print( 'saving result.')
np.savetxt('./result/%s.seg'%m.name,result,fmt='%d')
print("end at %s"%str(time.localtime()))