Method predict

model_assessment/clf.py:20–26

(self, threshold=0.5)

        self.confuse_matrix()

    def predict(self, threshold=0.5):
        # 根据阈值判断正类还是负类
        a = np.empty(shape=np.array(self.y_predict_rate).shape)
        for i in range(len(self.y_predict_rate)):
            if self.y_predict_rate[i] >= threshold: a[i] = 1
            else: a[i] = 0
        self.y_predict = a

    def confuse_matrix(self):
        # 计算混淆矩阵