brendanhasz / probflow

A Python package for building Bayesian models with TensorFlow or PyTorch
http://probflow.readthedocs.io
MIT License
171 stars 17 forks source link

Heteroskedastic pf.LinearRegression has no bias! #16

Closed brendanhasz closed 4 years ago

brendanhasz commented 4 years ago

For pf.applications.LinearRegression with heteroskedastic=True, the bias has shape [1, 1]. It should have shape [1, 2] and use the 2nd for the std dev bias - currently the std dev is missing a bias term!

brendanhasz commented 4 years ago

Fixed by https://github.com/brendanhasz/probflow/commit/e48ef518f1500e9b1735d3ed078c19eea56b35e2