Merge pull request #725 from hanzigs/master

Keras Sequential Model Support for Bias Variance Computation
rasbt · Sep 11, 2020 · a6ff5b8 · a6ff5b8
2 parents 8eb6954 + 81e7a87
commit a6ff5b8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/docs/sources/CHANGELOG.md b/docs/sources/CHANGELOG.md
@@ -19,7 +19,8 @@ The CHANGELOG for the current development version is available at
 
 ##### New Features
 
-- -
+- The `bias_variance_decomp` now supports Keras estimators. ([#725](https://github.com/rasbt/mlxtend/pull/725) via [@hanzigs](https://github.com/hanzigs))
+
 
 ##### Changes
 

diff --git a/mlxtend/evaluate/bias_variance_decomp.py b/mlxtend/evaluate/bias_variance_decomp.py
@@ -74,7 +74,11 @@ def bias_variance_decomp(estimator, X_train, y_train, X_test, y_test,
 
     for i in range(num_rounds):
         X_boot, y_boot = _draw_bootstrap_sample(rng, X_train, y_train)
-        pred = estimator.fit(X_boot, y_boot).predict(X_test)
+        if estimator.__class__.__name__ == 'Sequential':
+            estimator.fit(X_boot, y_boot)
+            pred = estimator.predict(X_test).reshape(1, -1)
+        else:
+            pred = estimator.fit(X_boot, y_boot).predict(X_test)
         all_pred[i] = pred
 
     if loss == '0-1_loss':