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 <npan...@us.ibm.com> Authored: Tue Oct 9 13:25:47 2018 -0700 Committer: Niketan Pansare <npan...@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