annotate more songs

[asr1617data.git] / train.py

diff --git a/train.py b/train.py
index 4d5e361..9d4d19a 100644 (file)
--- a/train.py
+++ b/train.py
@@ -4,9 +4,11 @@ from keras.layers import Dense, Dropout  # , Activation
 
 model = Sequential()
 
-model.add(Dense(50, input_shape=(13,), activation='relu'))
-model.add(Dense(50, activation='relu'))
-model.add(Dense(50, activation='relu'))
+model.add(Dense(26, input_shape=(13,), activation='relu'))
+model.add(Dense(100, activation='relu'))
+model.add(Dropout(0.25))
+model.add(Dense(100, activation='relu'))
+model.add(Dense(26, activation='relu'))
 model.add(Dense(1, activation='sigmoid'))
 
 model.compile(