update LSTM

sunlanchang · Jun 14, 2020 · 9c92d33 · 9c92d33
1 parent 8180cba
commit 9c92d33
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/LSTM_age_multi_input.py b/LSTM_age_multi_input.py
@@ -280,10 +280,12 @@ def save_data(datas):
 
 # %%
 try:
+    examples = args.examples
     mail('start train lstm')
     model.fit(
-        {'creative_id': x1_train, 'ad_id': x2_train, 'product_id': x3_train},
-        y_train,
+        {'creative_id': x1_train[:examples], 'ad_id': x2_train[:examples],
+            'product_id': x3_train[:examples]},
+        y_train[:examples],
         validation_data=([x1_val, x2_val, x3_val], y_val),
         epochs=args.epoch,
         batch_size=args.batch_size,