Skip to main content
Ctrl+K

Tabular Ensemble 0.3 documentation

Site Navigation

  • Get Started
  • Advanced Usage
  • API References

Site Navigation

  • Get Started
  • Advanced Usage
  • API References

Section Navigation

API references

  • tabensemb.trainer
    • tabensemb.trainer.Trainer
    • tabensemb.trainer.load_trainer
    • tabensemb.trainer.save_trainer
  • tabensemb.model
    • tabensemb.model.AbstractModel
    • tabensemb.model.TorchModel
    • tabensemb.model.AbstractNN
    • tabensemb.model.AutoGluon
    • tabensemb.model.WideDeep
    • tabensemb.model.PytorchTabular
    • tabensemb.model.CatEmbed
    • tabensemb.model.AbstractWrapper
    • tabensemb.model.TorchModelWrapper
    • tabensemb.model.WideDeepWrapper
    • tabensemb.model.PytorchTabularWrapper
    • tabensemb.model.RFE
  • tabensemb.data
    • tabensemb.data.base.AbstractDataStep
    • tabensemb.data.datamodule.DataModule
    • tabensemb.data.AbstractDeriver
    • tabensemb.data.dataderiver
      • tabensemb.data.dataderiver.get_data_deriver
      • tabensemb.data.dataderiver.RelativeDeriver
      • tabensemb.data.dataderiver.SampleWeightDeriver
      • tabensemb.data.dataderiver.UnscaledDataDeriver
    • tabensemb.data.AbstractImputer
    • tabensemb.data.AbstractSklearnImputer
    • tabensemb.data.dataimputer
      • tabensemb.data.dataimputer.get_data_imputer
      • tabensemb.data.dataimputer.GainImputer
      • tabensemb.data.dataimputer.MeanImputer
      • tabensemb.data.dataimputer.MedianImputer
      • tabensemb.data.dataimputer.MiceImputer
      • tabensemb.data.dataimputer.MiceLightgbmImputer
      • tabensemb.data.dataimputer.MissForestImputer
      • tabensemb.data.dataimputer.ModeImputer
    • tabensemb.data.AbstractProcessor
    • tabensemb.data.AbstractAugmenter
    • tabensemb.data.AbstractTransformer
    • tabensemb.data.AbstractFeatureSelector
    • tabensemb.data.AbstractScaler
    • tabensemb.data.dataprocessor
      • tabensemb.data.dataprocessor.get_data_processor
      • tabensemb.data.dataprocessor.CategoricalOrdinalEncoder
      • tabensemb.data.dataprocessor.CorrFeatureSelector
      • tabensemb.data.dataprocessor.FeatureValueSelector
      • tabensemb.data.dataprocessor.IQRRemover
      • tabensemb.data.dataprocessor.NaNFeatureRemover
      • tabensemb.data.dataprocessor.RFEFeatureSelector
      • tabensemb.data.dataprocessor.SampleDataAugmenter
      • tabensemb.data.dataprocessor.StandardScaler
      • tabensemb.data.dataprocessor.StdRemover
      • tabensemb.data.dataprocessor.VarianceFeatureSelector
    • tabensemb.data.AbstractSplitter
    • tabensemb.data.datasplitter
      • tabensemb.data.datasplitter.get_data_splitter
      • tabensemb.data.datasplitter.RandomSplitter
    • tabensemb.data.utils
      • tabensemb.data.utils.fill_cat_nan
      • tabensemb.data.utils.get_corr_sets
      • tabensemb.data.utils.get_imputed_dtype
      • tabensemb.data.utils.get_unknown_value
      • tabensemb.data.utils.OrdinalEncoder
  • tabensemb.config
    • tabensemb.config.UserConfig
  • tabensemb.utils
    • tabensemb.utils.utils
      • tabensemb.utils.utils.add_postfix
      • tabensemb.utils.utils.auto_metric_sklearn
      • tabensemb.utils.utils.check_stream
      • tabensemb.utils.utils.convert_proba_to_target
      • tabensemb.utils.utils.convert_target_to_indicator
      • tabensemb.utils.utils.debugger_is_active
      • tabensemb.utils.utils.get_figsize
      • tabensemb.utils.utils.gini
      • tabensemb.utils.utils.is_notebook
      • tabensemb.utils.utils.metric_sklearn
      • tabensemb.utils.utils.pretty
      • tabensemb.utils.utils.reload_module
      • tabensemb.utils.utils.safe_mkdir
      • tabensemb.utils.utils.seed_worker
      • tabensemb.utils.utils.set_random_seed
      • tabensemb.utils.utils.set_torch
      • tabensemb.utils.utils.str_to_dataframe
      • tabensemb.utils.utils.update_defaults_by_kwargs
      • tabensemb.utils.utils.HiddenPltShow
      • tabensemb.utils.utils.HiddenPrints
      • tabensemb.utils.utils.Logger
      • tabensemb.utils.utils.Logging
      • tabensemb.utils.utils.PickleAbleGenerator
      • tabensemb.utils.utils.PlainText
      • tabensemb.utils.utils.TqdmController
      • tabensemb.utils.utils.global_setting
      • tabensemb.utils.utils.torch_with_grad
    • tabensemb.utils.ranking
      • tabensemb.utils.ranking.avg_rank
      • tabensemb.utils.ranking.merge_leaderboards
      • tabensemb.utils.ranking.merge_to_excel
      • tabensemb.utils.ranking.read_lbs
  • API references
  • tabensemb.model
  • tabensemb.model.TorchModelWrapper

tabensemb.model.TorchModelWrapper#

class tabensemb.model.TorchModelWrapper(model: TorchModel)[source]#

Bases: AbstractWrapper

Methods

__init__(model: TorchModel)[source]#

reset_forward()

Reset the overridden forward method of the torch.nn.Module to ensure pickling compatibility.

wrap_forward()

Override the forward method of a torch.nn.Module to record hidden representations.

previous

tabensemb.model.AbstractWrapper

next

tabensemb.model.WideDeepWrapper
On this page
  • TorchModelWrapper
    • TorchModelWrapper.__init__()
Edit on GitHub
Show Source

© Copyright 2023, Tabular Ensemble developers.

Created using Sphinx 7.2.5.

Built with the PyData Sphinx Theme 0.13.3.