IndexError: Index dimension must be 1 or 2

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

X_trainがpandasのDataFrameであるため、LabelEncoderを適用する際に、エラーが発生しています。このエラーは、DataFrameの列を1次元のNumpy配列として渡す必要があるため発生しています。

修正するには、X_trainの各列を1次元のNumpy配列として抽出する必要があります。修正したコードは以下のようになります:

python
# LabelEncoder を使用して各列をエンコードする
label_encoders = {}
for col in categorical_cols:
    label_encoders[col] = LabelEncoder()
    X_train[col] = label_encoders[col].fit_transform(X_train[col].values)

この修正により、各列が1次元のNumpy配列として取り出され、LabelEncoderが適切に適用されるはずです。