dnn_predict.py 766 B

1234567891011121314151617181920212223242526272829303132
  1. import keras
  2. # -*- encoding:utf-8 -*-
  3. import numpy as np
  4. from keras.models import Sequential
  5. from keras.layers import Dense,Dropout
  6. import random
  7. from keras.models import load_model
  8. def read_data(path):
  9. lines = []
  10. with open(path) as f:
  11. for line in f.readlines()[:1000]:
  12. lines.append(eval(line.strip()))
  13. size = len(lines[0])
  14. train_x=[s[:size - 2] for s in lines]
  15. train_y=[s[size-1] for s in lines]
  16. return np.array(train_x),np.array(train_y)
  17. test_x,test_y=read_data("D:\\data\\quantization\\stock_test.log")
  18. path="model_seq.h5"
  19. model=load_model(path)
  20. score = model.evaluate(test_x, test_y)
  21. print(score)
  22. result=model.predict(test_x)
  23. # print(result)
  24. i = 0
  25. for x in test_y:
  26. # print(str(i) + ":" + str(x))
  27. i = i + 1