mlcompare.data.KaggleDataset

class mlcompare.data.KaggleDataset(**data)

Represents a Kaggle dataset with all the fields required to download and prepare it for model evaluation.

Attributes:

user (str): Username of the Kaggle user who owns the dataset. dataset (str): Name of the Kaggle dataset. file (str): Name of the file to be downloaded from the dataset. 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. If None, the file will be named user_dataset. 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)

• user (str)

• dataset (str)

• file (str)