Изменения

from sklearn.datasets import fetch_openml from numpy import arange import random from sklearn.tree import DecisionTreeClassifier from sklearn import datasets, svm, metrics

mnist = fetch_openml('MNIST_784') indices = arange(len(mnist.data)) randidx = random.sample(list(indices), 500) mnist.data = mnist.data[randidx] mnist.target = mnist.target[randidx] X = mnist.data Y = mnist.target train = len(X)//2 clf = DecisionTreeClassifier(criterion="entropy", max_depth=5) clf.fit(X[:train], Y[:train]) expected = Y[train:] predicted = clf.predict(X[train:]) print("Classification report for classifier %s:\n%s\n" % (clf, metrics.classification_report(expected, predicted)))