Repository: systemml Updated Branches: refs/heads/master c93d80602 -> 46f4d9207
[MINOR] Fix keras2dml tabs/spaces inconsistency Prevents conflict with some IDEs causing errors when mixing tabs and spaces Closes #763. Project: http://git-wip-us.apache.org/repos/asf/systemml/repo Commit: http://git-wip-us.apache.org/repos/asf/systemml/commit/46f4d920 Tree: http://git-wip-us.apache.org/repos/asf/systemml/tree/46f4d920 Diff: http://git-wip-us.apache.org/repos/asf/systemml/diff/46f4d920 Branch: refs/heads/master Commit: 46f4d920775cdef595a65bce4a134fe256cce28f Parents: c93d806 Author: Jeff Macaluso <[email protected]> Authored: Thu May 17 16:55:52 2018 -0700 Committer: Matthias Boehm <[email protected]> Committed: Thu May 17 16:55:52 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/46f4d920/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 de8aeb9..52fe9cc 100644 --- a/src/main/python/systemml/mllearn/estimators.py +++ b/src/main/python/systemml/mllearn/estimators.py @@ -946,7 +946,7 @@ class Keras2DML(Caffe2DML): self.name = keras_model.name createJavaObject(sparkSession._sc, 'dummy') if not hasattr(keras_model, 'optimizer'): - keras_model.compile(loss='categorical_crossentropy', optimizer=keras.optimizers.SGD(lr=0.01, momentum=0.95, decay=5e-4, nesterov=True)) + keras_model.compile(loss='categorical_crossentropy', optimizer=keras.optimizers.SGD(lr=0.01, momentum=0.95, decay=5e-4, nesterov=True)) convertKerasToCaffeNetwork(keras_model, self.name + ".proto", int(batch_size)) convertKerasToCaffeSolver(keras_model, self.name + ".proto", self.name + "_solver.proto", int(max_iter), int(test_iter), int(test_interval), int(display), lr_policy, weight_decay, regularization_type) self.weights = tempfile.mkdtemp() if weights is None else weights
