distfun_to_dist: Function to define output distribution based on dist_fun
Description
Function to define output distribution based on dist_fun
Usage
distfun_to_dist(dist_fun, preds)
Value
a symbolic tfp distribution
Arguments
- dist_fun
a distribution function as defined by make_tfd_dist
- preds
tensors with predictions