The predicted mean array returned by
|
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.