| #!/usr/bin/env python3 | |
| # -*- coding: utf-8 -*- | |
| import joblib | |
| import numpy as np | |
| from sklearn.datasets import load_iris | |
| from sklearn.model_selection import train_test_split | |
| from sklearn.naive_bayes import GaussianNB | |
| # Set the random seed | |
| random_seed = 0 | |
| np.random.seed(random_seed) | |
| # Load the dataset | |
| dataset = load_iris() | |
| X, y = dataset.data, dataset.target | |
| # Split the dataset into training and testing sets | |
| X_train, _, y_train, _ = train_test_split(X, y, test_size=0.25, random_state=random_seed) | |
| # Create and train model | |
| model = GaussianNB() | |
| model.fit(X_train, y_train) | |
| # Save the model to disk | |
| joblib.dump(model, 'gaussian_nb.joblib') | |