adriangb / scikeras

Scikit-Learn API wrapper for Keras.
https://www.adriangb.com/scikeras/
MIT License
239 stars 47 forks source link

Overwrite _fit function #292

Open gmbss0 opened 1 year ago

gmbss0 commented 1 year ago

The fit function that calls _fit expects numpy arrays or dataframes. For the project I am working on, it would be preferred to provide the training data as a tensorflow dataset. Does it make sense to overwrite the _fit function to allow the usage of tensorflow dataset or would it break to many dependencies?

adriangb commented 1 year ago

https://github.com/adriangb/scikeras/pull/166?