diff --git a/ML/aufgaben/naivebayes/naivebayes_3v8_variance.png b/ML/aufgaben/naivebayes/naivebayes_3v8_variance.png new file mode 100644 index 0000000..f33951f Binary files /dev/null and b/ML/aufgaben/naivebayes/naivebayes_3v8_variance.png differ diff --git a/ML/aufgaben/naivebayes/naivebayes_digits.py b/ML/aufgaben/naivebayes/naivebayes_digits.py index 1b70bae..07f6a80 100644 --- a/ML/aufgaben/naivebayes/naivebayes_digits.py +++ b/ML/aufgaben/naivebayes/naivebayes_digits.py @@ -47,3 +47,12 @@ for i, ax in enumerate(axes.flat): ax.axis('off') fig.suptitle('NB: Pixel variance per class') fig.savefig('naivebayes_digits_variance.png', dpi=150, bbox_inches='tight') + +# plot the variance difference between two commonly confused digits like 3 and 8 to see +# on which pixels nb relies to tell them apart +plt.figure() +diff = abs(classifier.var_[3] - classifier.var_[8]) +plt.imshow(diff.reshape(8, 8), cmap='hot') +plt.title('Variance difference: 3 vs 8') +plt.colorbar() +plt.savefig('naivebayes_3v8_variance.png', dpi=150, bbox_inches='tight')