labels=q_labels, duplicates="drop", precision=precision)
// Drop the column which has been discretized
new_X = new_X.drop([column_name], axis=1)
// explicitly convert the labels column to "str" type
new_X = new_X.astype(dtype={"{}_q_label".format(column_name): "str"})
return new_X
After Change
labels=q_labels, duplicates="drop", precision=precision)
// Drop the continuous feature column which has been discritized
new_X = new_X.drop([column_name], axis=1) if self.__drop_features else new_X
// explicitly convert the labels column to "str" type
new_X = new_X.astype(dtype={"{}_q_label".format(column_name): "str"})
return new_X