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()