search

flennerhag / mlens

ML-Ensemble – high performance ensemble learning
http://ml-ensemble.com
MIT License
843 stars 108 forks source link

Serialize mlens superlearner with KerasRegressor inside #133

Closed isaienkov closed 4 years ago

isaienkov commented 4 years ago

Is there any possibility to save superlearner with a KerasRegressor inside it? When I use ensemble without KerasRegressor it works fine like:

import pickle
pickle.dump(model, open(filename, 'wb'))

But if we have KerasRegressor in any layer it is not working.

flennerhag commented 4 years ago

Hi! This is a problem with TensorFlow not being serializable. It might help if you switch the Keras backend (#97)?