search

WhatAShot / ExcelFormer

The pioneering neural network surpassing extremely-tuned XGboost and Catboost on varied tabular datasets.
45 stars 7 forks source link

Support scikit-learn compatible #3

Closed arandomgoodguy closed 2 months ago

arandomgoodguy commented 1 year ago

Hi, great work!

Is there any chance for ExcelFormer to support scikit-learn fit API, so we can use something like ExcelFormerClassifier.fit().

In your another repo, TabCaps, model can be fitted by TabCapsClassifier.fit(), which is good.

It can help compare performance of multiple tabular models if they are all in one format, thanks!

jyansir commented 1 year ago

Hi, great work!

Is there any chance for ExcelFormer to support scikit-learn fit API, so we can use something like ExcelFormerClassifier.fit().

In your another repo, TabCaps, model can be fitted by TabCapsClassifier.fit(), which is good.

It can help compare performance of multiple tabular models if they are all in one format, thanks!

Hello, many thanks for your attention to our work! And yes, we do plan to reorganize ExcelFormer codes to support uniform APIs as machine learning package scikit-learn, our another work T2G-Former is also on schedule. The original codes for paper experiment reproduction will be preserved in the personal repo, and the uniform version will appear in our org repo.

We are currently busy in our ongoing research work and will inform you once code reorganization is finished, it won't be late.

arandomgoodguy commented 1 year ago

Thanks for your quick response. I look forward to hearing from you soon.