feat: add Select() step

ibis-project / ibis-ml

IbisML is a library for building scalable ML pipelines using Ibis.

https://ibis-project.github.io/ibis-ml/

Apache License 2.0

96 stars 13 forks source link

Open jitingxu1 opened 3 months ago

jitingxu1 commented 3 months ago

Add a Select step, opposite to Drop step.

Use it as column filters in the last-mile preprocessing.

import ibis_ml as ml

recipe = ml.Recipe(
    # to be added
    ml.Select(ml.numeric()),
    ml.ImputeMean(ml.numeric()),
)

Any thought?

deepyaman commented 2 months ago

Sure, makes sense if we have Drop().