hp.py 文件源码

def sumIntensitiesAll(
        self,
        t,
        node_vec,
        etimes,
        filterlatertimes=False,
        ):
        if filterlatertimes:
            I = self.mu * np.sum(self.gamma) \
                + np.dot(np.transpose(self.alpha[node_vec[etimes
                         < t].astype(int), :][:, range(self.D)]),
                         self.kernel_evaluate(t, etimes[etimes < t],
                         self.omega))
        else:
            I = self.mu * np.sum(self.gamma) \
                + np.dot(np.transpose(self.alpha[node_vec.astype(int), :
                         ][:, range(self.D)]), self.kernel_evaluate(t,
                         etimes, self.omega))
        sumI = np.sum(I)
        return (I, sumI)