predict.py

   1 import numpy as np
   2 import sys
   3 from keras.models import model_from_json
   4
   5 with open('model.json', 'r') as f:
   6     json = f.read()
   7
   8 model = model_from_json(json)
   9 model.load_weights('./model.hdf5')
  10 model.compile(
  11     loss='binary_crossentropy',
  12     optimizer='rmsprop',
  13     metrics=['accuracy'])
  14
  15 dat = np.genfromtxt(sys.stdin.buffer, dtype=float, delimiter='\t')
  16 for i in model.predict(dat, batch_size=32, verbose=0):
  17     print(i[0])