yufeng 4 years ago
parent
commit
d775ee4291
1 changed files with 2 additions and 2 deletions
  1. 2 2
      stock/dnn_train_dmi.py

+ 2 - 2
stock/dnn_train_dmi.py

@@ -67,11 +67,11 @@ def resample(path, suffix='test'):
67 67
 
68 68
 
69 69
 def mul_train(name="10_18"):
70
-    for x in range(0, 12):
70
+    for x in range(2, 12):
71 71
     # for x in [11,0,1,3,8,9]:
72 72
     # for x in [11,0,1,3,5,6,8,9]:
73 73
         score = train(input_dim=440, result_class=5, file_path=data_dir + "stock"+ name + "_train_" + str(x) + ".log",
74
-              model_name='18d_dnn_seq_' + str(x) + '.h5')
74
+              model_name='11_18d_dnn_seq_' + str(x) + '.h5')
75 75
 
76 76
         with open(data_dir + 'stock' + name + '_dmi.log', 'a') as f:
77 77
             f.write(str(x) + ':' + str(score[1]) + '\n')