jaxfit.loss_functions module

class jaxfit.loss_functions.LossFunctionsJIT[source]

Bases: object

arctan(z, cost_only)[source]

cauchy(z, cost_only)[source]

construct_all_loss_functions()[source]

construct_single_loss_function(loss)[source]

create_arctan_funcs()[source]

create_calculate_cost()[source]

create_cauchy_funcs()[source]

create_huber_funcs()[source]

create_scale_rhos()[source]

create_soft_l1_funcs()[source]

create_stack_rhos()[source]

create_zscale()[source]

get_empty_rhos(z)[source]

get_loss_function(loss)[source]

huber(z, cost_only)[source]

soft_l1(z, cost_only)[source]