create_family_torch: Function to create (custom) family

View source: R/families_torch.R

create_family_torchR Documentation

Function to create (custom) family

Description

Function to create (custom) family

Usage

create_family_torch(torch_dist, trafo_list, output_dim = 1L)

Arguments

torch_dist

a torch 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 to train a distribution learning model in torch


deepregression documentation built on Sept. 9, 2025, 5:27 p.m.