tabensemb.model.WideDeepWrapper#
- class tabensemb.model.WideDeepWrapper(model: WideDeep)[source]#
Bases:
AbstractWrapperMethods
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.