# Y_train, Y_test = Y_train[::100], Y_test[::100] sess = tf.Session() sess.run(tf.global_variables_initializer()) # Training classifier = DBN(output_act_func='softmax', hidden_act_func='relu', loss_fuc='cross_entropy', use_for='classification', dbn_lr=1e-3, dbn_epochs=100, dbn_struct=[784, 100, 100,10], rbm_h_type='bin', rbm_epochs=10, batch_size=32, cd_k=1, rbm_lr=1e-3) classifier.build_dbn() classifier.train_dbn(X_train, Y_train,sess) # Test Y_pred = classifier.test_dbn(X_test, Y_test,sess) #import matplotlib.pyplot as plt #PX=range(0,len(Y_pred)) #plt.figure(1) # 选择图表1 #plt.plot(PX, Y_test,'r') #plt.plot(PX, Y_pred,'b') #plt.show()