回答編集履歴
1
コードが間違っていたので修正しました。(elifをifに直しました)
test
CHANGED
|
@@ -30,7 +30,7 @@
|
|
|
30
30
|
|
|
31
31
|
|
|
32
32
|
|
|
33
|
-
|
|
33
|
+
if (flatten):
|
|
34
34
|
|
|
35
35
|
x_train = x_train.reshape(len(x_train), -1)
|
|
36
36
|
|
|
@@ -38,7 +38,7 @@
|
|
|
38
38
|
|
|
39
39
|
|
|
40
40
|
|
|
41
|
-
|
|
41
|
+
if (one_hot_label):
|
|
42
42
|
|
|
43
43
|
y_train = to_categorical(y_train)
|
|
44
44
|
|