私はOneHotEncoder
自分のプロジェクトで Scikit-learn のを使用しています。n_value
そして、が に設定されている場合、各ワンホット ベクトルのサイズを知る必要がありますauto
。私はそれを示すと思っn_value_
たが、訓練サンプルを試す以外に方法がないようだ. 問題を示すために、このおもちゃのサンプル コードを作成しました。他の解決策を知っていますか?
from sklearn.preprocessing import OneHotEncoder
data = [[1], [3], [5]] # 3 different features
encoder = OneHotEncoder()
encoder.fit(data)
print(len(encoder.transform([data[0]]).toarray()[0])) # 3 number of dimensions in one-hot-vector
print(encoder.n_values_) # [6] == len(range(5))