参数调整下

stock/dnn_train.py

@@ -12,7 +12,7 @@ from imblearn.over_sampling import RandomOverSampler
 def read_data(path):
     lines = []
     with open(path) as f:
-        for x in range(100000):
+        for x in range(60000):
             lines.append(eval(f.readline().strip()))
 
     random.shuffle(lines)
@@ -42,9 +42,9 @@ def train(input_dim=400, result_class=3, file_path="D:\\data\\quantization\\stoc
     model = Sequential()
     model.add(Dense(units=120+input_dim, input_dim=input_dim,  activation='relu'))
     # model.add(Dense(units=60+int(input_dim/2), activation='relu'))
-    model.add(Dense(units=60+input_dim, activation='relu',kernel_regularizer=regularizers.l2(0.01)))
+    model.add(Dense(units=120+input_dim, activation='relu',kernel_regularizer=regularizers.l2(0.001)))
     model.add(Dropout(0.2))
-    model.add(Dense(units=60+input_dim, activation='relu',kernel_regularizer=regularizers.l2(0.01)))
+    model.add(Dense(units=60+input_dim, activation='relu'))
     model.add(Dropout(0.2))
     model.add(Dense(units=60+input_dim, activation='selu'))
     # model.add(Dropout(0.2))