We should convert the EnergyGap and EnergyZScore methods to use the joint_energy instead of the marginal_free_energy because the former is well-defined for deep models whereas the latter is not really computable. In fact, we should probably deprecate the marginal_free_energy function, which may simplify the code a bit.
This is really easy to fix, so we should just do it.
We should convert the
EnergyGap
andEnergyZScore
methods to use thejoint_energy
instead of themarginal_free_energy
because the former is well-defined for deep models whereas the latter is not really computable. In fact, we should probably deprecate the marginal_free_energy function, which may simplify the code a bit.This is really easy to fix, so we should just do it.