larry._utils._task_two_eval_callback

Module Contents

Classes

TaskTwoEvalCallback

Helper class that provides a standard way to create an ABC using

class larry._utils._task_two_eval_callback.TaskTwoEvalCallback(model, UMAP_model=None, t=torch.Tensor([2, 4, 6]), device=autodevice.AutoDevice(), N=2000, use_key='X_scaled', PR_threshold=0.3)

Bases: lightning.Callback, larry._utils._abc_parse.ABCParse

Helper class that provides a standard way to create an ABC using inheritance.

property _TESTING
property t
property fate_df
property t0_idx
property F_obs
property labels
property F_hat
property F_obs_pred
property LOGPATH
_plot_fate_bias_clustermap()
on_fit_end(trainer, pl_module, *args, **kwargs)