python
1from sklearn.datasets import load_wine 2from sklearn.ensemble import RandomForestClassifier 3from sklearn.metrics import accuracy_score 4from sklearn.model_selection import train_test_split 5data = load_wine() 6x_train, x_test, y_train, y_test = train_test_split(data.data, data.target, test_size = 0.3) 7model =RandomForestClassifier() 8model.fit(x_train, y_train) 9plt.barh(data.feature_names, model.feature_importances_) 10
このようなランダムフォレストの重要度のバーグラフを、重要度別にソートして出力したいので、
model.feature_importances_をソートした順番に紐づけて、data.feature_namesも一緒にソートしたいのですが、
プログラミング初心者のため、どうやってソートするのが一番シンプルか教えていただきたいです。
model.feature_importances_ と data.feature_names で一度データフレームを作成したほうがいいのか?
それとも、何か一緒にソートできる便利な関数があれば教えてください。
回答2件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。