Noumida commited on
Commit
0ea4ef6
·
verified ·
1 Parent(s): 9ef3ae0

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -0
model.py CHANGED
@@ -23,6 +23,8 @@ from sklearn.model_selection import cross_val_score
23
  from tensorflow.keras.callbacks import ModelCheckpoint
24
  from tensorflow.keras.models import Model, load_model
25
  from sklearn.model_selection import train_test_split
 
 
26
  model = Sequential()
27
  model.add(GRU(units=256, dropout=0.05, recurrent_dropout=0.35, return_sequences=True, input_shape=input_shape))
28
  model.add(GRU(units=768, dropout=0.05, recurrent_dropout=0.35, return_sequences=False))
 
23
  from tensorflow.keras.callbacks import ModelCheckpoint
24
  from tensorflow.keras.models import Model, load_model
25
  from sklearn.model_selection import train_test_split
26
+
27
+ input_shape=(64,40)
28
  model = Sequential()
29
  model.add(GRU(units=256, dropout=0.05, recurrent_dropout=0.35, return_sequences=True, input_shape=input_shape))
30
  model.add(GRU(units=768, dropout=0.05, recurrent_dropout=0.35, return_sequences=False))