123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778 |
- # -*- encoding:utf-8 -*-
- import numpy as np
- from keras.models import load_model
- def read_data(path):
- lines = []
- with open(path) as f:
- for line in f.readlines()[:]:
- lines.append(eval(line.strip()))
- size = len(lines[0])
- train_x=[s[:size - 2] for s in lines]
- train_y=[s[size-1] for s in lines]
- return np.array(train_x),np.array(train_y),lines
- def predict(file_path='', model_path='15min_dnn_seq.h5'):
- test_x,test_y,lines=read_data(file_path)
- model=load_model(model_path)
- score = model.evaluate(test_x, test_y)
- print('DNN', score)
- up_num = 0
- up_right = 0
- up_error = 0
- i = 0
- result=model.predict(test_x)
- win_dnn = []
- with open('dnn_predict_5d.txt', 'a') as f:
- for r in result:
- fact = test_y[i]
- if r[0] > 0.8:
- # f.write(str([lines[i][-2], lines[i][-1]]) + "\n")
- # win_dnn.append([lines[i][-2], lines[i][-1]])
- if fact[0] == 1:
- up_right = up_right + 1.1
- elif fact[1] == 1:
- up_error = up_error + 0.4
- up_right = up_right + 1.03
- else:
- up_error = up_error + 1
- up_right = up_right + 0.9
- up_num = up_num + 1
- elif r[1] > 0.5:
- pass
- # f.write(str([lines[i][-2], lines[i][-1]]) + "\n")
- # win_dnn.append([lines[i][-2], lines[i][-1]])
- # if fact[0] == 1:
- # up_right = up_right + 1.1
- # elif fact[1] == 1:
- # up_right = up_right + 1.03
- # # elif fact[2] == 1:
- # # up_right = up_right + 1
- # else:
- # up_right = up_right + 0.9
- # up_num = up_num + 1
- i = i + 1
- if up_num == 0:
- up_num = 1
- print('DNN', up_right, up_num, up_right/up_num, up_error/up_num)
- return win_dnn,up_right/up_num
- def multi_predict():
- r = 0
- for x in [0]:
- win_dnn, ratio = predict(file_path='D:\\data\\quantization\\kmeans\\stock2_10_' + str(x) + '_test.log', model_path='5d_dnn_seq_' + str(x) + '.h5')
- r = r + ratio
- print(r)
- if __name__ == '__main__':
- # predict(file_path='D:\\data\\quantization\\stock6_5_test.log', model_path='5d_dnn_seq.h5')
- predict(file_path='D:\\data\\quantization\\stock578A_12d_train3.log', model_path='5d_578A_dnn_seq.h5')
- # multi_predict()
|