New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Flatten the means from BaeysianNeuralFieldEstimator.predict #17

Open

fsaad opened this issue Feb 9, 2024 · 0 comments

Collaborator

fsaad commented Feb 9, 2024 •

edited

Loading

The predicted mean array returned by

bayesnf/src/bayesnf/spatiotemporal.py

Line 273 in 2debb69

def predict(self, table, quantiles=(0.5,)):

has three dimensions, (num_devices, ensemble_size // num_devices, len(table) (for MLE/MAP) or even four dimensions (for VI).

It is more convenient to have only two dimensions.

Related #19.

The text was updated successfully, but these errors were encountered:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment