多目的変数 ランダムフォレスト回帰 Boruta 特徴量選択

Question

入力特徴量を削減したくborutaを使おうとしているのですがエラーが出ます
よく分からないので助けてください。お願いします。

x_train=df_f.values
t_train=df_rslt.values

random_forest = RandomForestRegressor()

feat_selector = BorutaPy(random_forest, n_estimators='auto')  
#特徴量選択
feat_selector.fit(x_train, t_train)
mask = feat_selector.support_

pd.to_pickle(mask, "select_mask.pkl")#保存

print(mask)
train_selected_x = df_f.iloc[:,mask].values

エラー内容

Traceback (most recent call last):
File "select_feature.py", line 217, in <module>
feat_selector.fit(x_train, t_train)
File "/Users/KO/opt/anaconda3/lib/python3.7/site-packages/boruta/boruta_py.py", line 201, in fit
return self._fit(X, y)
File "/Users/KO/opt/anaconda3/lib/python3.7/site-packages/boruta/boruta_py.py", line 251, in _fit
self._check_params(X, y)
File "/Users/KO/opt/anaconda3/lib/python3.7/site-packages/boruta/boruta_py.py", line 517, in _check_params
X, y = check_X_y(X, y)
File "/Users/KO/opt/anaconda3/lib/python3.7/site-packages/sklearn/utils/validation.py", line 724, in check_X_y
y = column_or_1d(y, warn=True)
File "/Users/KO/opt/anaconda3/lib/python3.7/site-packages/sklearn/utils/validation.py", line 760, in column_or_1d
raise ValueError("bad input shape {0}".format(shape))
ValueError: bad input shape (79083, 15)

上記のようなエラーが出ていて、(79083, 15)は教師データの形で確認しました.
15通りの回帰です

rf_before = RandomForestRegressor()

# 実行!!
rf_before.fit(x_train, t_train)

borutaとは関係のなく上記のようにRandomForestRegressorのみでこの入力データを使って回帰させるとエラーなく動きます。

borutaは多目的変数の回帰はできない感じですか？

Answer 1

https://github.com/scikit-learn-contrib/boruta_py/blob/master/boruta/boruta_py.py#L197

yとして複数の目的変数は想定されていないようです。

Comments

[unchi]

なるほどそうでしたか
ありがとうございました