diff options
Diffstat (limited to 'lenet.py')
-rw-r--r-- | lenet.py | 3 |
1 files changed, 2 insertions, 1 deletions
@@ -64,7 +64,8 @@ def get_lenet(shape): model.add(Dense(units=120, activation='relu')) model.add(Dense(units=84, activation='relu')) - model.add(Dense(units=10, activation = 'softmax')) + #model.add(Dense(units=10, activation = 'softmax')) + model.add(Dense(units=10, activation = 'relu')) return model def plot_history(history, metric = None): |