DL之RBM:基于RBM实现手写数字图片识别提高准确率
生活随笔
收集整理的這篇文章主要介紹了
DL之RBM:基于RBM实现手写数字图片识别提高准确率
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
DL之RBM:基于RBM實現手寫數字圖片識別提高準確率
?
?
目錄
輸出結果
設計代碼
?
?
?
輸出結果
?
?
設計代碼
import numpy as np import matplotlib.pyplot as pltfrom sklearn.model_selection import train_test_split from sklearn import metrics,linear_model from sklearn.neural_network import BernoulliRBM from sklearn.datasets import load_digits from sklearn.pipeline import Pipeline digits = load_digits() X = digits.data y = digits.target X -= X.min() X /= X.max() X_train, X_test, y_train, y_test = train_test_split(X, y) logistic = linear_model.LogisticRegression() rbm = BernoulliRBM(random_state=0, verbose=True) classifier = Pipeline(steps=[('rbm', rbm), ('logistic',logistic)]) rbm.learning_rate = 0.06 rbm.n_iter = 20 rbm.n_components = 200 logistic.C = 6000.0 classifier.fit (X_train,y_train) print() print("Logistic regression using RBM features:\n%s\n"%(metrics.classification_report(y_test,classifier.predict(X_test))?
總結
以上是生活随笔為你收集整理的DL之RBM:基于RBM实现手写数字图片识别提高准确率的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: DL之BM:BM的前世今生
- 下一篇: TF之LSTM:基于Tensorflow