tabensemb.model.RFE._prepare_tensors#

method

RFE._prepare_tensors(df, derived_data, model_name)#

Transform the upcoming dataset into Tensors that has the same structures as those stored in a tabensemb.data.datamodule.DataModule and obtained by tabensemb.data.datamodule.DataModule.update_dataset().

Parameters:

df: The new tabular dataset that has the same structure as self.trainer.datamodule.X_test
derived_data: Unstacked data derived from tabensemb.data.datamodule.DataModule.derive_unstacked().
model_name: The name of the selected model.

Returns:

A tuple of torch.Tensor: Transformed tensors.
pd.DataFrame: The transformed dataset after running _run_custom_data_module().
dict: The derived unstacked data after running _run_custom_data_module()
DataModule: The tabensemb.data.datamodule.DataModule returned by _prepare_custom_data_module()