Pyhton实现决策树算法 MNIST数据集

Pyhton实现决策树算法 MNIST数据集

决策树是一种比较接近人类思维方式的算法,将样本通过每个特征值的信息增益进行划分,从而保证每个划分之后的结果信息熵的消减量达到最大。具体的原理请大家自己查找相关资料。

sklearn实现代码如下, 准确率可以达到90%左右。

from sklearn import tree
import numpy as np
from tensorflow.examples.tutorials.mnist import input_data

mnist = input_data.read_data_sets('MNIST_data', one_hot=Fal
  • 1
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
抱歉,我是一个自然语言处理模型,无法提供代码功能。但是,我可以为您提供一些参考信息来实现您的要求。 使用决策树算法实现MNIST数据集的读取和显示、数据集的划分、算法实现、结果预测需要以下步骤: 1. 读取和显示MNIST数据集:MNIST数据集可以通过Python中的一些库来读取和显示,例如numpy,matplotlib。您可以使用以下代码读取并显示MNIST数据集: ``` import numpy as np import matplotlib.pyplot as plt from sklearn.datasets import fetch_openml # Load MNIST dataset mnist = fetch_openml('mnist_784') # Split dataset into features and labels X, y = mnist.data, mnist.target # Reshape data to 28x28 images X = X.reshape((-1, 28, 28)) # Display first image in the dataset plt.imshow(X[0], cmap='gray') plt.title('Label: {}'.format(y[0])) plt.show() ``` 2. 数据集的划分:您可以使用sklearn库中的train_test_split函数将数据集划分为训练集和测试集。以下是一个示例代码: ``` from sklearn.model_selection import train_test_split # Split dataset into training and testing sets X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) ``` 3. 决策树算法实现:您可以使用sklearn库中的DecisionTreeClassifier类来实现决策树算法。以下是一个示例代码: ``` from sklearn.tree import DecisionTreeClassifier # Create decision tree classifier clf = DecisionTreeClassifier() # Train classifier on training data clf.fit(X_train.reshape((-1, 28*28)), y_train) ``` 4. 结果预测:使用训练好的决策树模型对测试数据进行预测,并计算准确率。以下是一个示例代码: ``` from sklearn.metrics import accuracy_score # Predict labels for testing data y_pred = clf.predict(X_test.reshape((-1, 28*28))) # Calculate accuracy of predictions accuracy = accuracy_score(y_test, y_pred) print('Accuracy: {:.2f}%'.format(accuracy*100)) ``` 以上是使用决策树算法实现MNIST数据集的读取和显示、数据集的划分、算法实现、结果预测的基本步骤和示例代码。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值