diff --git a/ML/aufgaben/naivebayes/naivebayes_iris.py b/ML/aufgaben/naivebayes/naivebayes_iris.py index d614729..bbe3000 100644 --- a/ML/aufgaben/naivebayes/naivebayes_iris.py +++ b/ML/aufgaben/naivebayes/naivebayes_iris.py @@ -28,6 +28,7 @@ classifier.fit(x_train, y_train) score = classifier.score(x_test, y_test) print(score) +# plot the distributions (bell curves) for each feature and each flower fig, axes = plt.subplots(2, 2, figsize=(12, 10)) for idx, ax in enumerate(axes.flat): x_range = np.linspace(iris.data[:, idx].min() - 1, iris.data[:, idx].max() + 1, 200)