Update model.py
Browse files
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))
|