torch_frame.transforms.MutualInformationSort

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

基础类:FittableBaseTransform

A transform that sorts the numerical features of input

基于互信息的TensorFrame对象。

Parameters:
  • task_type (TaskType) – 任务类型。

  • na_strategy (NAStrategy) – 用于填补数值特征中NaN值的策略。