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']]