Learn practical skills, build real-world projects, and advance your career
!pip install jovian --upgrade --quiet
WARNING: pip is being invoked by an old script wrapper. This will fail in a future version of pip. Please see https://github.com/pypa/pip/issues/5599 for advice on fixing the underlying issue. To avoid this problem you can invoke Python with '-m pip' instead of running pip directly.
import matplotlib.pyplot as plt
from sklearn.datasets import load_iris
iris = load_iris()

import pandas as pd
X = pd.DataFrame(iris.data)
y = pd.DataFrame(iris.target)

import numpy as np
colormap = np.array(['red', 'lime', 'black'])
plt.figure(figsize=(14,7))

from sklearn.cluster import KMeans
model = KMeans(n_clusters=3)
model.fit(X)
plt.subplot(1, 2, 2)
plt.scatter(X[2],X[3], c=colormap[model.labels_])
plt.title('K Mean Classification')

import sklearn.metrics as sm
print(sm.accuracy_score(y, model.labels_))

from sklearn.mixture import GaussianMixture
gmm = GaussianMixture(n_components=3)
gmm.fit(X)
y_cluster_gmm=gmm.predict(X)
plt.subplot(1, 2, 1)
plt.scatter(X[2],X[3], c=colormap[y_cluster_gmm])
plt.title('GMM Classification')

print(sm.accuracy_score(y, y_cluster_gmm))
print(sm.confusion_matrix(y, y_cluster_gmm))
0.24 0.3333333333333333 [[ 0 50 0] [45 0 5] [ 0 0 50]]
Notebook Image
jovian