溫馨提示×

Matplotlib怎么與scikit-learn聯(lián)合使用

小億
88
2024-05-21 14:30:26
欄目: 編程語言

Matplotlib是一個(gè)用于繪制圖形的Python庫,而scikit-learn是一個(gè)用于機(jī)器學(xué)習(xí)的Python庫。你可以使用Matplotlib來可視化scikit-learn中的數(shù)據(jù)和模型。

下面是一個(gè)簡單的例子,展示如何使用Matplotlib和scikit-learn一起工作:

import matplotlib.pyplot as plt
from sklearn import datasets
from sklearn.cluster import KMeans

# 加載數(shù)據(jù)集
iris = datasets.load_iris()
X = iris.data
y = iris.target

# 使用KMeans算法進(jìn)行聚類
kmeans = KMeans(n_clusters=3)
kmeans.fit(X)
y_kmeans = kmeans.predict(X)

# 將聚類結(jié)果可視化
plt.scatter(X[:, 0], X[:, 1], c=y_kmeans, cmap='viridis')
centers = kmeans.cluster_centers_
plt.scatter(centers[:, 0], centers[:, 1], c='red', s=200, alpha=0.5)
plt.xlabel('Sepal length')
plt.ylabel('Sepal width')
plt.show()

在這個(gè)例子中,我們首先加載了鳶尾花數(shù)據(jù)集,并使用KMeans算法將數(shù)據(jù)分為3個(gè)簇。然后,我們使用Matplotlib繪制了數(shù)據(jù)點(diǎn)和聚類中心點(diǎn)的散點(diǎn)圖,以可視化聚類結(jié)果。

通過結(jié)合使用Matplotlib和scikit-learn,你可以更好地理解和展示機(jī)器學(xué)習(xí)模型的結(jié)果。你可以根據(jù)需要來調(diào)整圖形的樣式和參數(shù),以獲得更好的可視化效果。

0