teratail header banner
teratail header banner
質問するログイン新規登録

質問編集履歴

1

ショートコードがエラーになるのを修正

2017/08/25 09:13

投稿

iwahara
iwahara

スコア6

title CHANGED
File without changes
body CHANGED
@@ -21,12 +21,18 @@
21
21
  ショートコードですが。
22
22
 
23
23
  ```python
24
+ from keras.models import Sequential
25
+ from keras.layers.core import Dense, Activation, Dropout
26
+ from keras.optimizers import SGD
27
+ import numpy as np
28
+ from keras.utils import np_utils
29
+
24
30
  model = Sequential()
25
31
  model.add(Dense(64, activation='relu', input_dim=3))
26
32
  model.add(Dropout(0.5))
27
33
  model.add(Dense(64, activation='relu'))
28
34
  model.add(Dropout(0.5))
29
- model.add(Dense(nb_classes, activation='softmax'))
35
+ model.add(Dense(3, activation='softmax'))
30
36
  sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True)
31
37
  model.compile(loss='categorical_crossentropy',
32
38
  optimizer=sgd,