python - 将 CSV 数据导入 scikit-learn？

Question

我将数据存储在 CSV 文件中，如下所示:

abv,ibu,label
6.1,60.0,American Pale Ale (APA)
9.9,92.0,American Barleywine
7.9,45.0,Winter Warmer
4.4,42.0,American Pale Ale (APA)`

我的理解是它接受二维数组中的数据。

Answer 1

import pandas as pd
dat = pd.read_csv('file.csv')

在使用 scikit-learn 之前，您需要对标签进行 one-hot 编码。

y = pd.get_dummies(dat['label'])
X = dat[:2]

这些是总体思路，希望对您有所帮助。