tune embedding size

lgstd · Apr 15, 2016 · 200ec1c · 200ec1c
1 parent 944ea5f
commit 200ec1c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models.py b/models.py
@@ -195,8 +195,8 @@ def __build_keras_model(self):
         models = []
 
         model_store = Sequential()
-        model_store.add(Embedding(1115, 50, input_length=1))
-        model_store.add(Reshape(dims=(50,)))
+        model_store.add(Embedding(1115, 10, input_length=1))
+        model_store.add(Reshape(dims=(10,)))
         models.append(model_store)
 
         model_dow = Sequential()