DL之DNN：利用DNN算法对mnist手写数字图片识别数据集(sklearn自带,1797*64)训练、预测(95%)

你个睿智 2022-09-20 09:12:25  50244

分类专栏：资讯

数据集展示

输出结果

设计代码

数据集展示

先查看sklearn自带digits手写数据集(1797*64)

输出结果

设计代码


import numpy as np
from sklearn.datasets import load_digits 
from sklearn.metrics import confusion_matrix, classification_report 
from sklearn.preprocessing import LabelBinarizer 
from NeuralNetwork import NeuralNetwork
from sklearn.cross_validation import train_test_split 
digits = load_digits() 
X = digits.data  
y = digits.target
X -= X.min() 
X /= X.max()
nn = NeuralNetwork([64, 100, 10], 'logistic')  
 
X_train, X_test, y_train, y_test = train_test_split(X, y)  
labels_train = LabelBinarizer().fit_transform(y_train)
labels_test = LabelBinarizer().fit_transform(y_test)
print ("start fitting")
nn.fit(X_train, labels_train, epochs=3000) 
predictions = [] 
for i in range(X_test.shape[0]): 
    o = nn.predict(X_test[i])           
    predictions.append(np.argmax(o))    
print (confusion_matrix(y_test, predictions) )      
print (classification_report(y_test, predictions) )

网站声明：如果转载，请联系本站管理员。否则一切后果自行承担。

本文链接：https://www.xckfsq.com/news/show.html?id=3884

赞同 0

评论 0 条

DL之DNN：利用DNN算法对mnist手写数字图片识别数据集(sklearn自带,1797*64)训练、预测(95%)

数据集展示

输出结果

设计代码

相关文章

关注我们