diff --git a/code/artificial_intelligence/src/gaussian_naive_bayes/gaussian_native_bayes.py b/code/artificial_intelligence/src/gaussian_naive_bayes/gaussian_native_bayes.py new file mode 100644 index 0000000000..3272410168 --- /dev/null +++ b/code/artificial_intelligence/src/gaussian_naive_bayes/gaussian_native_bayes.py @@ -0,0 +1,18 @@ +from sklearn.datasets import load_iris +from sklearn.model_selection import train_test_split +from sklearn.naive_bayes import GaussianNB +from sklearn.metrics import accuracy_score + +data = load_iris() +X = data.data +y = data.target + +X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) + +model = GaussianNB() +model.fit(X_train, y_train) + +y_pred = model.predict(X_test) + +accuracy = accuracy_score(y_test, y_pred) +print(f'Accuracy: {accuracy * 100:.2f}%')