美文网首页
使用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