torch_frame.transforms.MutualInformationSort

class MutualInformationSort(task_type: TaskType, na_strategy: NAStrategy = NAStrategy.MEAN)[source]

Bases: FittableBaseTransform

A transform that sorts the numerical features of input

TensorFrame object based on mutual information.

Parameters:

  • task_type (TaskType) – The task type.

  • na_strategy (NAStrategy) – Strategy used for imputing NaN values in numerical features.