predict.py 506 B

12345678910111213141516171819202122232425262728293031
  1. from keras.models import load_model
  2. import numpy as np
  3. from keras.utils import np_utils
  4. def read_data(path):
  5. with open(path) as f :
  6. lines=f.readlines()[0:3]
  7. lines=[eval(line.strip()) for line in lines]
  8. X,Y=zip(*lines)
  9. X=np.array(X)
  10. X=1.0*X/256
  11. X=X.reshape(-1,28*28)
  12. Y=np.array(Y)
  13. Y=np_utils.to_categorical(Y,num_classes)
  14. return X,Y
  15. num_classes=11
  16. X,Y=read_data("test_data")
  17. model = load_model('model')
  18. results=model.predict(X)
  19. for result,y in zip(results,Y):
  20. print(result)
  21. print(y)