Label EncodingとOne-Hot Encodingのサンプルコード

2024年6月8日2024年6月17日

以下に、Label EncodingとOne-Hot Encodingのサンプルコードを提供します。

Label Encodingのサンプルコード：

python
from sklearn.preprocessing import LabelEncoder

# LabelEncoderを作成して、メーカー品番をエンコードする
label_encoder = LabelEncoder()
X_maker_number_encoded = label_encoder.fit_transform(X_maker_number)

# エンコードされたメーカー品番をデータフレームに追加する
X_train['X_maker_number_encoded'] = X_maker_number_encoded

One-Hot Encodingのサンプルコード：

python
from sklearn.preprocessing import OneHotEncoder

# OneHotEncoderを作成して、メーカー品番をエンコードする
onehot_encoder = OneHotEncoder(sparse=False)
X_maker_number_encoded = onehot_encoder.fit_transform(X_maker_number.values.reshape(-1, 1))

# エンコードされたメーカー品番をデータフレームに追加する
X_maker_number_encoded_df = pd.DataFrame(X_maker_number_encoded, columns=onehot_encoder.get_feature_names_out(['X_maker_number']))
X_train = pd.concat([X_train, X_maker_number_encoded_df], axis=1)

このように、Label EncodingはLabelEncoderを使用し、One-Hot EncodingはOneHotEncoderを使用して行います。それぞれのエンコーディング方法で新しい特徴量を作成し、元のデータフレームに追加します。