tabensemb.model.CatEmbed._generate_dataset_from_tensors#

method

CatEmbed._generate_dataset_from_tensors(tensors, df, derived_data, model_name)#

Perform the same preprocessing as in _generate_dataset() on a new dataset.

Parameters:

tensors: Tensors that has the same structures as those stored in a tabensemb.data.datamodule.DataModule and obtained by tabensemb.data.datamodule.DataModule.update_dataset().
df: The transformed dataset after running _run_custom_data_module().
derived_data: The derived unstacked data after running _run_custom_data_module()
model_name: The name of the selected model.

Returns: