aaa6b9c9c0a803153ae26a3b28fe2c7fbf3e75fe,paysage/metrics.py,EnergyGap,update,#EnergyGap#Any#,239

Before Change

        
        self.norm += 1
        self.energy_gap += be.mean(update_args.model
                                   .joint_energy(update_args.minibatch))
        self.energy_gap -= be.mean(update_args.model
                                   .joint_energy(update_args.random_samples))

    def value(self):

After Change

            None

        
        energy_data = update_args.model.joint_energy(update_args.minibatch)
        energy_random = update_args.model.joint_energy(update_args.random_samples)
        self.calc.update(energy_data)
        self.calc.update(-energy_random)

Instances