treeflow.vi.util module

class treeflow.vi.util.VIResults(loss, parameters)

Bases: tuple

count(value, /)

Return number of occurrences of value.

index(value, start=0, stop=9223372036854775807, /)

Return first index of value.

Raises ValueError if the value is not present.

loss

Alias for field number 0

parameters

Alias for field number 1

treeflow.vi.util.default_vi_trace_fn(traceable_quantities: MinimizeTraceableQuantities, variables_dict: Dict[str, Variable]) → VIResults