HeteroscedasticSNGPLayer error with num_classes=2
Open
- Dominant language
- Jupyter Notebook
- Stars
- 712
- Forks
- 77
- Avg merge
- 9h 39m
- Merged PRs (30d)
- 1
Description
HeteroscedasticSNGPLayer inherits from MCSoftmaxDenseFA which has an assert check:
https://github.com/google/edward2/blob/2a852121f5e3c47c88e4212aae86d8ba33febdbe/edward2/tensorflow/layers/heteroscedastic.py#L605
but itself is written in a way that should accept num_classes=2:
https://github.com/google/edward2/blob/2a852121f5e3c47c88e4212aae86d8ba33febdbe/edward2/tensorflow/layers/hetsngp.py#L232
Contributor guide
Assessment
This issue has not been assessed yet.