plotting in rbm train_batch