if config.LAST_LAYER == "softmax":
for predictlabel in X:
predict_new = []
prob = np.argsort(predictlabel)
pred = [prob[-1]]
total_prob = config.SOFTMAX_THRESHHOLD - prob[-1]
for i in range(len(prob))[2:]:
total_prob -= prob[-i]
if total_prob > 0:
pred.append(prob[i - 1])
else:
break
xitem = np.array(pred)
predict_new.append(list(xitem))
predict_labels_list.extend(predict_new)
predict_label_and_marked_label_list = zip(predict_labels_list, marked_labels_list)
score = judger.get_taskaccu_score(predict_label_and_marked_label_list)
print('阈值：%.4f'%threshold)
print('分数：%.6f'%score)
print(time.time() - time1)
if score>lastscore:
lastscore = score
lastthreshold = threshold
print('最优阈值：%.4f'%lastthreshold)
print('最优分数：%.6f'%lastscore)
print(time.time() - time0)
# break
sys.exit()
