cas-pml/ML/unterlagen/08_NB_Digits.ipynb

In [1]:

from sklearn import datasets
digits = datasets.load_digits()
print(digits.data.shape)

(1797, 64)

In [2]:

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(digits.data, digits.target, test_size=0.2, random_state=0)
print(X_train.shape)
print(y_train.shape)

(1437, 64)
(1437,)

In [3]:

from sklearn.naive_bayes import GaussianNB
nb = GaussianNB()
nb.fit(X_train, y_train)
nb.score(X_test,y_test)

Out [3]:

0.82499999999999996

In [4]:

from sklearn.tree import DecisionTreeClassifier
dt = DecisionTreeClassifier()
dt.fit(X_train, y_train)
dt.score(X_test,y_test)

Out [4]:

0.85833333333333328

In [5]:

from sklearn.naive_bayes import GaussianNB
from sklearn.model_selection import cross_val_score
nb2 = GaussianNB()
scores = cross_val_score(nb2, digits.data, digits.target, cv=10)
print("Accuracy 0.95 confidence interval: %0.2f (+/- %0.2f)" % (scores.mean(), scores.std() * 2))

Accuracy 0.95 confidence interval: 0.81 (+/- 0.11)

In [ ]: