RNN-G: make train_dict for graphing training