Replace n_features_ with n_features_in_ in TSBF example

johannfaouzi · Oct 28, 2021 · f971b27 · f971b27
1 parent 936e092
commit f971b27
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/classification/plot_tsbf.py b/examples/classification/plot_tsbf.py
@@ -56,10 +56,10 @@
 clf = TSBF(random_state=43, bins=5)
 clf.fit(X_train, y_train)
 
-plt.bar(np.arange(clf.n_features_), clf.feature_importances_)
+plt.bar(np.arange(clf.n_features_in_), clf.feature_importances_)
 plt.title('Feature importance scores')
-plt.xticks(np.arange(clf.n_features_),
-           ['feature {}'.format(i) for i in range(clf.n_features_)],
+plt.xticks(np.arange(clf.n_features_in_),
+           ['feature {}'.format(i) for i in range(clf.n_features_in_)],
            rotation=90)
 plt.ylabel("Mean decrease in impurity")
 plt.tight_layout()