You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@systemml.apache.org by ni...@apache.org on 2018/10/09 20:26:40 UTC

systemml git commit: [MINOR] Bugfix in Keras2DML API when loading weights from Keras

Repository: systemml
Updated Branches:
  refs/heads/master ef15e582b -> 8a144f2b3


[MINOR] Bugfix in Keras2DML API when loading weights from Keras

Project: http://git-wip-us.apache.org/repos/asf/systemml/repo
Commit: http://git-wip-us.apache.org/repos/asf/systemml/commit/8a144f2b
Tree: http://git-wip-us.apache.org/repos/asf/systemml/tree/8a144f2b
Diff: http://git-wip-us.apache.org/repos/asf/systemml/diff/8a144f2b

Branch: refs/heads/master
Commit: 8a144f2b35343a7aa8fbb4bf7aedd31dd36a3852
Parents: ef15e58
Author: Niketan Pansare <np...@us.ibm.com>
Authored: Tue Oct 9 13:25:47 2018 -0700
Committer: Niketan Pansare <np...@us.ibm.com>
Committed: Tue Oct 9 13:25:47 2018 -0700

----------------------------------------------------------------------
 src/main/python/systemml/mllearn/estimators.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/systemml/blob/8a144f2b/src/main/python/systemml/mllearn/estimators.py
----------------------------------------------------------------------
diff --git a/src/main/python/systemml/mllearn/estimators.py b/src/main/python/systemml/mllearn/estimators.py
index d231b08..fbcd3e2 100644
--- a/src/main/python/systemml/mllearn/estimators.py
+++ b/src/main/python/systemml/mllearn/estimators.py
@@ -1017,7 +1017,7 @@ class Keras2DML(Caffe2DML):
         weight_decay: regularation strength (default: 5e-4)
         regularization_type: regularization type (default: "L2")
         """
-        from .keras2caffe import convertKerasToCaffeNetwork, convertKerasToCaffeSolver
+        from .keras2caffe import convertKerasToCaffeNetwork, convertKerasToCaffeSolver, convertKerasToSystemMLModel
         import tempfile, keras
         if isinstance(keras_model, keras.models.Sequential):
             # Convert the sequential model to functional model