search

sktime / mlaut

Other
24 stars 5 forks source link

Subclass keras estimator #12

Closed ViktorKaz closed 6 years ago

ViktorKaz commented 6 years ago

the predict() method of keras sequential models seems to defaults to predict_proba(). This introduces inconsistency which can be fixed by subclassing the kearas estimator class and implementing predict().

ViktorKaz commented 6 years ago

This is implemented. See estimators.nn_estimators. In particular, see the OverwrittenSequentialClassifier class.