From 63963aced31782b2c3296df7382273bb50da148a Mon Sep 17 00:00:00 2001 From: sunlanchang Date: Mon, 15 Jun 2020 22:43:22 +0800 Subject: [PATCH] update LSTM --- LSTM_gender_multi_input.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/LSTM_gender_multi_input.py b/LSTM_gender_multi_input.py index d937681..d510df9 100644 --- a/LSTM_gender_multi_input.py +++ b/LSTM_gender_multi_input.py @@ -283,7 +283,7 @@ def save_npy(datas, name): np.save(f'tmp/{name}_{i}.npy', data) inputs = [DATA['X1_train'], DATA['X1_val'], DATA['X2_train'], - DATA['X2_val'], DATA['X3_train'], DATA['x3_val']] + DATA['X2_val'], DATA['X3_train'], DATA['X3_val']] targets = [DATA['Y_train'], DATA['Y_val']] embeddings = [DATA['creative_id_emb'], DATA['ad_id_emb'], DATA['product_id_emb']]