2

すべての整数がカテゴリに対応する整数のベクトルがあるとします。

A = [1, 2, 2, 3, 3, 1, 2, 4, 4, 1]

私は自分がいくつのカテゴリーを持っているかを知っています。このベクトルは、ロジスティック回帰モデルで終了する X データセットの列の 1 つです。

sciki-tlearn 関数onehotencoderを使用して次のようなものを取得することは可能ですか:

0 0 0 1 (1のとき)
0 0 1 0 (2のとき)
0 1 0 0 (3のとき)
1 0 0 0 (なんでも)

またはさらに良い

0 0 0
0 0 1
0 1 0
1 0 0

?

そのようなベクトルを onehotencoder に渡そうとすると、次のエラーが発生しますneed more than 1 value to unpack

さらに、「NULL」レコードがある場合、最初にそれらを数値に変換する必要があると思いA(find(A=='NULL'))=123ます。

ご協力ありがとうございました。フランチェスコ

4

1 に答える 1