cuvis_ai.supervised.skorch_supervised.SkorchSupervised

class cuvis_ai.supervised.skorch_supervised.SkorchSupervised(epochs: int = 10, optimizer: Union[cuvis_ai.utils.nn_config.Optimizer, torch.optim.optimizer.Optimizer] = None, verbose: bool = False, model: torch.nn.modules.module.Module = None, model_args: dict = <factory>)

Bases: BaseSupervised

__init__(epochs: int = 10, optimizer: ~cuvis_ai.utils.nn_config.Optimizer | ~torch.optim.optimizer.Optimizer | None = None, verbose: bool = False, model: ~torch.nn.modules.module.Module | None = None, model_args: dict = <factory>) → None

Methods

__init__([epochs, optimizer, verbose, ...])
fit(X, Y)
forward(X)
load()
serialize()

Attributes

epochs
model
optimizer
verbose
model_args

epochs: int = 10

fit(X: ndarray, Y: ndarray)

forward(X: ndarray)

load()

model: Module = None

model_args: dict

optimizer: Optimizer | Optimizer = None

serialize()

verbose: bool = False