Python sklearn load_wine用法及代碼示例

本文簡要介紹python語言中 sklearn.datasets.load_wine 的用法。

用法:

sklearn.datasets.load_wine(*, return_X_y=False, as_frame=False)

加載並返回 wine 數據集(分類)。

葡萄酒數據集是一個經典且非常簡單的多類分類數據集。

Classes	3
每班樣本	[59,71,48]
樣品總數	178
Dimensionality	13
Features	真實的，積極的

在用戶指南中閱讀更多信息。

參數：

return_X_y：布爾，默認=假

如果為 True，則返回 (data, target) 而不是 Bunch 對象。有關data 和target 對象的更多信息，請參見下文。

as_frame：布爾，默認=假

如果為 True，則數據是 pandas DataFrame，包括具有適當數據類型(數字)的列。目標是 pandas DataFrame 或 Series，具體取決於目標列的數量。如果 return_X_y 為 True，則 ( data , target ) 將是 pandas DataFrames 或 Series，如下所述。

返回：

data：sklearn.utils.Bunch

類似字典的對象，具有以下屬性。

數據：{ndarray, dataframe} 形狀 (178, 13)

數據矩陣。如果as_frame=True，data將是一個pandas DataFrame。

目標：{ndarray, Series} 形狀 (178,)

分類目標。如果as_frame=True，target將是一個pandas係列。

feature_names：列表

數據集列的名稱。

target_names：列表

目標類的名稱。

框架：DataFrame 形狀 (178, 14)

僅在 as_frame=True 時出現。 DataFrame 與 data 和 target 。

說明：str

數據集的完整說明。

(data, target)：如果return_X_y 為真，則為元組

UCI ML Wine Data Set 數據集的副本已下載並修改以適應

標準格式來自：

https://archive.ics.uci.edu/ml/machine-learning-databases/wine/wine.data

例子：

假設您對樣本 10、80 和 140 感興趣，並且想知道它們的類名。

>>> from sklearn.datasets import load_wine
>>> data = load_wine()
>>> data.target[[10, 80, 140]]
array([0, 1, 2])
>>> list(data.target_names)
['class_0', 'class_1', 'class_2']