Skip to content

Commit

Permalink
update LSTM
Browse files Browse the repository at this point in the history
  • Loading branch information
sunlanchang committed Jun 15, 2020
1 parent cee5977 commit 63963ac
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion LSTM_gender_multi_input.py
Original file line number Diff line number Diff line change
Expand Up @@ -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']]
Expand Down

0 comments on commit 63963ac

Please sign in to comment.