GAMPTeam / vampyre

Approximate Message Passing in Python
MIT License
65 stars 28 forks source link

fix infinite variance handling in GaussEst, DiscreteEst, LinEst and ReLUEst #13

Open julianwi opened 1 week ago

julianwi commented 1 week ago

The infinite variance handling in GaussEst and DiscreteEst was broken, because it passed avg_var_cost into the ind_out parameter of est_init().

The infinite variance code path in ReLUEst was broken as well, because zhat0_map and related variables did not get set in this code path.

Infinite variance handling was also added to LinEst and the unit tests for all four estimators have been extended to test the infinite variance cases.