catenets.models.torch.utils.model_utils module

Model utils shared across different nets

make_val_split(X: torch.Tensor, y: torch.Tensor, w: Optional[torch.Tensor] = None, val_split_prop: float = 0.3, seed: int = 42, stratify_w: bool = True) → Any

predict_wrapper(estimator: Any, X: torch.Tensor) → torch.Tensor

train_wrapper(estimator: Any, X: torch.Tensor, y: torch.Tensor, **kwargs: Any) → None