mlchem.ml.preprocessing.scaling.transform_df

transform_df(df: DataFrame, scaler: StandardScaler | MinMaxScaler | RobustScaler, last_columns_to_preserve: int) tuple[DataFrame, StandardScaler | MinMaxScaler | RobustScaler]

Transform a DataFrame using a provided scaler, preserving specified columns.

Parameters:
  • df (pandas.DataFrame) – The input DataFrame.

  • scaler (StandardScaler or MinMaxScaler or RobustScaler) – The fitted scaler to use for transformation.

  • last_columns_to_preserve (int) – Number of columns at the end of the DataFrame to exclude from transformation.

Returns:

The transformed DataFrame and the scaler used.

Return type:

tuple of pandas.DataFrame and scaler