15a58be0f54f22ea43b8db25d11dd0c402acf47f,paysage/metrics.py,EnergyZscore,update,#EnergyZscore#,323

Before Change

            None

        
        self.data_mean += be.mean(update_args.model
                                  .joint_energy(update_args.minibatch))
        self.random_mean += be.mean(update_args.model
                                     .joint_energy(update_args.random_samples))
        self.random_mean_square += be.mean(update_args.model
                                           .joint_energy(update_args.random_samples)**2)

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_data.update(energy_data)
        self.calc_random.update(energy_random)
        // self.data_mean += be.mean(update_args.model

Instances