diff options
author | Vasil Zlatanov <vasil@netcraft.com> | 2019-03-06 23:49:46 +0000 |
---|---|---|
committer | Vasil Zlatanov <vasil@netcraft.com> | 2019-03-06 23:49:46 +0000 |
commit | 5d779afb5a9511323e3402537af172d68930d85c (patch) | |
tree | c31d546c7759c53b23948e170d690e727a295810 /lenet.py | |
parent | b418990448f461da50a732b4e66dd8e9066199d8 (diff) | |
download | e4-gan-5d779afb5a9511323e3402537af172d68930d85c.tar.gz e4-gan-5d779afb5a9511323e3402537af172d68930d85c.tar.bz2 e4-gan-5d779afb5a9511323e3402537af172d68930d85c.zip |
Replace softmax with relu as we apply it in the function anyway
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): |