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

Before Change

        // drop null columns
        data.dropna(axis=1, how="all", inplace=True)

        if self.strategy == "zero":
            for col in data.columns:
                data[col].fillna(value=0,inplace=True)                
            return data
        elif self.strategy == "ignore_row":
            data.dropna(axis=0, how="any", inplace=True)
            return data
        elif self.strategy == "ignore_column":
            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"
            raise TypeError(msg)

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