tabensemb.model.PytorchTabularWrapper#
- class tabensemb.model.PytorchTabularWrapper(model: PytorchTabular)[source]#
Bases:
AbstractWrapperMethods
- __init__(model: PytorchTabular)[source]#
Reset the overridden forward method of the torch.nn.Module to ensure pickling compatibility.
Override the forward method of a torch.nn.Module to record hidden representations.