tabensemb.model.AbstractNN.teardown

method

AbstractNN.teardown(stage: str) → None

Called at the end of fit (train + validate), validate, test, or predict.

Parameters:

stage – either 'fit', 'validate', 'test', or 'predict'