create_family {deepregression}

Function to create (custom) family

Description

Function to create (custom) family

Usage

create_family(tfd_dist, trafo_list, output_dim = 1L)

Arguments

tfd_dist

a tensorflow probability distribution

trafo_list

list of transformations h for each parameter (e.g, exp for a variance parameter)

output_dim

integer defining the size of the response

Value

a function that can be used by tfp$layers$DistributionLambda to create a new distribuional layer