mlcompare.data.BaseDataset

class mlcompare.data.BaseDataset(**data)

Base class for datasets, containing attributes related to data cleaning and reformatting.

Attributes:

target (str): Column name for the target of the predictions. save_name (str | None): Name to use for files saved from this dataset. Should be unique across datasets. drop (list[str] | None): List of column names to be dropped from the dataset. one_hot_encode (list[str] | None): List of column names to be one-hot encoded in the dataset.

Parameters:

• target (str)

• saveName (str | None)

• drop (list[str] | None)

• nan (Literal['ffill', 'bfill', 'drop'] | None)

• ordinalEncode (list[str] | None)

• oneHotEncode (list[str] | None)

• targetEncode (list[str] | None)

• labelEncode (Literal['yes'] | None)

• standardScale (list[str] | None)

• minMaxScale (list[str] | None)

• maxAbsScale (list[str] | None)

• robustScale (list[str] | None)

• powerTransform (list[str] | None)

• quantileTransform (list[str] | None)

• quantileTransformNormal (list[str] | None)

• normalize (list[str] | None)