Diff of /create_rnn_model.py [000000] .. [6ab127]

Switch to side-by-side view

--- a
+++ b/create_rnn_model.py
@@ -0,0 +1,11 @@
+from tensorflow.keras.models import Sequential
+from tensorflow.keras.layers import Embedding, LSTM, Dense
+
+def create_rnn_model(input_shape, num_classes):
+    model = Sequential()
+    model.add(Embedding(input_dim=10000, output_dim=128, input_length=input_shape[1]))
+    model.add(LSTM(64))
+    model.add(Dense(128, activation='relu'))
+    model.add(Dense(num_classes, activation='softmax'))
+    
+    return model