diff --git a/constants.py b/constants.py index 4e4e6d1..41ad4f1 100644 --- 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') \ No newline at end of file diff --git a/predict.py b/predict.py index 1dc4da2..ca418c6 100755 --- 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 index d05ae2a..f5cfab3 100755 --- 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):