..

durablenapkin · Jun 28, 2018 · 0b79559 · 0b79559
1 parent 8bd0e73
commit 0b79559
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/constants.py b/constants.py
@@ -1,6 +1,6 @@
 import os
 
-BATCH_SIZE = os.getenv('BATCH_SIZE', 8)
+BATCH_SIZE = os.getenv('BATCH_SIZE', 256)
 MAX_STRING_LEN = 255
 MODEL = os.getenv('MODEL', 'model.h5')
 WEIGHTS = os.getenv('WEIGHTS', 'weights.h5')
diff --git a/predict.py b/predict.py
@@ -16,6 +16,9 @@
 
 from constants import MAX_STRING_LEN, MODEL, WEIGHTS, BATCH_SIZE
 
+BATCH_SIZE = int(BATCH_SIZE)
+MAX_STRING_LEN = int(MAX_STRING_LEN)
+
 
 def predict(i):
     tokenizer = Tokenizer(filters='\t\n', char_level=True, lower=True)

diff --git a/train.py b/train.py
@@ -20,8 +20,8 @@
 BATCH_SIZE = int(BATCH_SIZE)
 MAX_STRING_LEN = int(MAX_STRING_LEN)
 
-NEURONS = os.getenv('TF_NEURONS', 128)  # g00gle.com 256?
-EMBEDDED_DIM = 300
+NEURONS = os.getenv('TF_NEURONS', 128)
+EMBEDDED_DIM = 500
 
 
 def train(csv_file):