f0a4b4d03fd700c96800dd1d68a2cc0dfc182f83,cheml/preprocessing/handle_missing.py,missing_values,transform,#missing_values#Any#,99

Before Change

            data.dropna(axis=1, how="any", inplace=True)
            return data
        elif self.strategy == "interpolate":
            data = data.interpolate()
            data.fillna(method="ffill",axis=1, inplace=True) // because of nan in the first and last element of column
            return data
        else:
            msg = "Wrong strategy has been passed"

After Change

        if self.strategy == "ignore_row":
            return df[self.mask]
        elif self.strategy == "ignore_column":
            return df.T[self.mask].T
        else:
            msg = "The transform method doesn"t change the dataframe if strategy="zero" or "interpolate". You should fit_transform your new dataframe with those methods."
            warnings.warn(msg)

Instances