Update plot_rbm_logistic_classification.py