create_family_torch {deepregression} | R Documentation |
Function to create (custom) family
create_family_torch(torch_dist, trafo_list, output_dim = 1L)
torch_dist |
a torch probability distribution |
trafo_list |
list of transformations h for each parameter
(e.g, |
output_dim |
integer defining the size of the response |
a function that can be used to train a distribution learning model in torch