机器学习-标签数值映射

最新推荐文章于 2024-03-13 22:07:36 发布

原创

最新推荐文章于 2024-03-13 22:07:36 发布 · 2.2k 阅读

6 ·

CC 4.0 BY-SA版权

Santorinisu博客，未经授权，禁止转载!!

文章标签：

#机器学习 #python

本文介绍了如何将无顺序的Nominal标签转换为数值类型，包括使用OneHotEncoder和LabelEncoder的方法，以及Pandas的get_dummies功能。在处理非排序标签时，需要注意转换后结果的稀疏性和额外维度。

Section I: Map Ordinary Categorical Label

第一部分：代码

import pandas as pd
import numpy as np

df=pd.DataFrame([['green','M',10.1,'class1'],
                 ['red','L',13.5,'class2'],
                 ['blue','XL',15.3,'class1']])

#Section 1: Encoding class labels via class-integer mapping dictionary
df.columns=['color','size','price','classlabel']
class_mapping={
   
   label:idx for idx,label in enumerate(np.unique(df['classlabel']))}

print("Transform Before:\n",df)
df['classlabel']=df['classlabel'].map(class_mapping)
print("After Transform:\n"