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 2019/03/23 02:48:50 UTC

[systemml] branch gh-pages updated: [MINOR][DOC] Updated Deep Learning documentation

This is an automated email from the ASF dual-hosted git repository.

niketanpansare pushed a commit to branch gh-pages
in repository https://gitbox.apache.org/repos/asf/systemml.git


The following commit(s) were added to refs/heads/gh-pages by this push:
     new 2eed56f  [MINOR][DOC] Updated Deep Learning documentation
2eed56f is described below

commit 2eed56fd88bbeb633dfa6d452d55110594ff310d
Author: Niketan Pansare <np...@us.ibm.com>
AuthorDate: Fri Mar 22 19:47:00 2019 -0700

    [MINOR][DOC] Updated Deep Learning documentation
    
    - Also, fixed javadoc errors.
---
 deep-learning.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deep-learning.md b/deep-learning.md
index 2dbb4bb..968c959 100644
--- a/deep-learning.md
+++ b/deep-learning.md
@@ -207,6 +207,7 @@ keras_model.add(Flatten())
 keras_model.add(Dense(512, activation='relu'))
 keras_model.add(Dropout(0.5))
 keras_model.add(Dense(10, activation='softmax'))
+keras_model.compile(loss='categorical_crossentropy', optimizer=SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True))
 keras_model.summary()
 
 # Scale the input features