torch_mist.utils.evaluation

Submodules

• torch_mist.utils.evaluation.mi_estimator
• torch_mist.utils.evaluation.model

Package Contents

Functions

evaluate(, batch_size, num_workers, Dict[str, float]])
evaluate_mi(, batch_size, num_workers, Dict[str, float]])

torch_mist.utils.evaluation.evaluate(model: torch.nn.Module, method: str, data: torch_mist.utils.data.utils.TensorDictLike, device: torch.device = torch.device('cpu'), batch_size: int | None = None, num_workers: int = 0) → float | Dict[str, float]

torch_mist.utils.evaluation.evaluate_mi(estimator: torch_mist.estimators.base.MIEstimator, data: torch_mist.utils.data.utils.TensorDictLike, device: torch.device = torch.device('cpu'), batch_size: int | None = None, num_workers: int = 0) → float | Dict[str, float]