SJ001 / AI-Feynman

MIT License
607 stars 186 forks source link

sklearn-style interface? #60

Open asross opened 2 years ago

asross commented 2 years ago

I was thinking about trying this method for a project, but the interface for running it (directories of text files and CSVs) seems pretty tricky to use / non-standard for Python users -- most people have their input data in numpy or pandas files, and don't want to parse through a complicated directory structure to inspect results.

It seems like it would be pretty straightforward, however, to create a scikit-learn-style wrapper class with a fit(X,y) method that:

Overall, I think creating an easy-to-use API for this method would do a ton to increase its impact. I was able to get other libraries like PySR and gplearn running in minutes, but this feels like it would take a good bit longer to figure out / set up.