ertk.pytorch.classification.pt_train_val_test

ertk.pytorch.classification.pt_train_val_test(model_fn: Callable[[...], Module | LightningModule], train_data: tuple[numpy.ndarray, ...], valid_data: tuple[numpy.ndarray, ...], test_data: tuple[numpy.ndarray, ...] | None = None, scoring: str | list[str] | dict[str, collections.abc.Callable[[numpy.ndarray, numpy.ndarray], float]] | Callable[[...], float] = 'accuracy', verbose: int = 0, fit_params: dict[str, Any] = {}) → ExperimentResult