search

google / bayesnf

Bayesian Neural Field models for prediction in large-scale spatiotemporal datasets
https://google.github.io/bayesnf/
Apache License 2.0
93 stars 10 forks source link

Make approximate_quantiles an argument to predict #18

Closed fsaad closed 9 months ago

fsaad commented 9 months ago

Currently they are hardcoded:

https://github.com/google/bayesnf/blob/2debb69cf21daf7de982843a82436a57d5a97dd3/src/bayesnf/spatiotemporal.py#L282 https://github.com/google/bayesnf/blob/2debb69cf21daf7de982843a82436a57d5a97dd3/src/bayesnf/spatiotemporal.py#L362

Then remove predict from BayesianNeuralFieldMLE.