distfun_to_dist {deepregression}

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)

Arguments

dist_fun

a distribution function as defined by make_tfd_dist

preds

tensors with predictions

Value

a symbolic tfp distribution