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 <npan...@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