tabensemb.data.datamodule.DataModule.generate_tensors#
method
- DataModule.generate_tensors(scaled_df, derived_data)[source]#
Generate PyTorch tensors.
- Parameters:
- scaled_df
The scaled dataset.
- derived_data
A dict of derived unstacked data calculated by
derive_unstacked()
- Returns:
- torch.Tensor
A tensor of continuous features, a list of tensors of derived_unstacked data, and a tensor of the target.