HBLR.py 文件源码

def predictProbability(self, task, X):
        prob = 0
        for k in range(self.K):
            numerator = np.dot(np.atleast_2d(self.theta[k,:]),X.T)
            diag_entries = np.einsum('ij,ij->i', np.dot(X, self.gamma[k]), X) ##
            denom = np.sqrt(1.0 + np.pi/8 * diag_entries)
            prob = prob + self.phi[task,k] * sigmoid(numerator / denom)
        return prob


    # Code for Predicting for a new task