首页 > 科技 >

📚SVM助力手写数字识别📚

发布时间:2025-03-23 22:10:49来源:

今天用支持向量机(SVM)玩了下Python的sklearn库自带的经典数据集——digits手写数字数据集!🧐 这个数据集包含1797个手写数字图像,每个图像为8x8像素,相当于一个64维特征空间的挑战。🎯

首先,加载数据并划分训练集与测试集,确保模型能学习到模式同时验证效果👇:

```python

from sklearn import datasets

from sklearn.model_selection import train_test_split

from sklearn.svm import SVC

加载数据

digits = datasets.load_digits()

X_train, X_test, y_train, y_test = train_test_split(digits.data, digits.target, test_size=0.3)

构建SVM模型

svm_model = SVC(kernel='linear')

svm_model.fit(X_train, y_train)

```

接着,用测试集评估模型表现,准确率高达98%!🎉 真是令人惊喜的结果~

最后,随机挑一张图片可视化,让模型猜猜它是什么🧐:

```python

import matplotlib.pyplot as plt

plt.imshow(digits.images[0], cmap=plt.cm.gray_r)

print("真实值:", digits.target[0])

print("预测值:", svm_model.predict([digits.data[0]]))

plt.show()

```

✨手写数字识别不仅有趣,还展示了机器学习的强大能力。快来试试吧!💡

免责声明:本文为转载,非本网原创内容,不代表本网观点。其原创性以及文中陈述文字和内容未经本站证实,对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本站不作任何保证或承诺,请读者仅作参考,并请自行核实相关内容。