美文网首页
使用LogisticRegression逻辑回归完成MNIST的

使用LogisticRegression逻辑回归完成MNIST的

作者: Sara_d94e | 来源:发表于2020-09-22 00:49 被阅读0次

MNIST 数据集是经典的手写数字识别数据集,每个样本28*28

精简版MNIST:一共1797个样本,每个样本8*8,sklearn自带

完整版MNIST:一共6万个样本(5万个训练,1万个测试),每个样本28*28

http://deeplearning.net/data/mnist/mnist.pkl.gz

# 引入包

from sklearn.model_selection import train_test_split

from sklearn.linear_model import LogisticRegression

from sklearn import preprocessing

from sklearn.metrics import accuracy_score

from sklearn.datasets import load_digits

# 加载数据

digits = load_digits()

data = digits.data

# 数据预处理

# 采用Z-Score规范化

ss = preprocessing.StandardScaler()

train_ss_x = ss.fit_transform(train_x)

test_ss_x = ss.transform(test_x)

# 创建LR分类器

lr = LogisticRegression()

lr.fit(train_ss_x, train_y)

predict_y=lr.predict(test_ss_x)

print('LR准确率: %0.4lf' % accuracy_score(test_y, predict_y))

LR准确率: 0.9644

相关文章

网友评论

      本文标题:使用LogisticRegression逻辑回归完成MNIST的

      本文链接:https://www.haomeiwen.com/subject/gsjiyktx.html