我有一个通过数据框加载的数据集,其中类别标签需要使用scikit-learn中的LabelEncoder
进行编码。列label
是类别标签列,其包含以下类别:
[‘Standing’, ‘Walking’, ‘Running’, ‘null’]
为了执行标签编码,我尝试了以下方法但没有成功。如何修复它?
from sklearn import preprocessingimport pandas as pddf = pd.read_csv('dataset.csv', sep=',') df.apply(preprocessing.LabelEncoder().fit_transform(df['label']))
回答:
你可以尝试如下方法:
le = preprocessing.LabelEncoder()df['label'] = le.fit_transform(df.label.values)
或者以下方法也可以工作:
df['label'] = le.fit_transform(df['label'])
这将用编码后的标签替换数据框中原始的label
值。