f8121ee825e731a0bf3b3fd8d411b5926c89ab74,pyhawkes/internals/weights.py,GammaMixtureWeights,expected_W,#GammaMixtureWeights#,283

Before Change

        Compute the expected W under the variational approximation
        
        p_A = self.expected_A()
        return p_A * self.expected_W_given_A(1.0) + (1-p_A) * self.expected_W_given_A(0.0)

    def expected_W_given_A(self, A):
        
        Compute the expected W given A under the variational approximation

After Change

        
        p_A = self.expected_A()

        E_W =  p_A * self.expected_W_given_A(1.0) + (1-p_A) * self.expected_W_given_A(0.0)

        if not self.network.allow_self_connections:
            np.fill_diagonal(E_W, 0.0)
        return E_W

    def expected_W_given_A(self, A):
        
        Compute the expected W given A under the variational approximation

Instances