Better xlabel in plot_importance() (#11009)

dmlc · Nov 21, 2024 · 2e189a8 · 2e189a8
1 parent a502f67
commit 2e189a8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python-package/xgboost/plotting.py b/python-package/xgboost/plotting.py
@@ -24,7 +24,7 @@ def plot_importance(
     xlim: Optional[tuple] = None,
     ylim: Optional[tuple] = None,
     title: str = "Feature importance",
-    xlabel: str = "F score",
+    xlabel: str = "Importance score",
     ylabel: str = "Features",
     fmap: PathLike = "",
     importance_type: str = "weight",

diff --git a/tests/python/test_plotting.py b/tests/python/test_plotting.py
@@ -28,7 +28,7 @@ def test_plotting(self):
         ax = xgb.plot_importance(booster)
         assert isinstance(ax, Axes)
         assert ax.get_title() == 'Feature importance'
-        assert ax.get_xlabel() == 'F score'
+        assert ax.get_xlabel() == 'Importance score'
         assert ax.get_ylabel() == 'Features'
         assert len(ax.patches) == 4
 

diff --git a/tests/python/test_with_sklearn.py b/tests/python/test_with_sklearn.py
@@ -660,7 +660,7 @@ def test_sklearn_plotting():
     ax = xgb.plot_importance(classifier)
     assert isinstance(ax, Axes)
     assert ax.get_title() == 'Feature importance'
-    assert ax.get_xlabel() == 'F score'
+    assert ax.get_xlabel() == 'Importance score'
     assert ax.get_ylabel() == 'Features'
     assert len(ax.patches) == 4