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

tests: test DiscretizeKBins and CreatePolynomialFeatures on different backends #99

Closed jitingxu1 closed 6 months ago

jitingxu1 commented 6 months ago

Hi,

I am not sure if this is the right way to test Step on different backends. I just use this as the start point for collecting better solutions and feedbacks. I tried this solution on two Step: DiscretizeKBins and CreatePolynomialFeatures

The main idea is to see if the operators used in fit_table() and transform_table could be compiled to sql of different backend.