Merge pull request scikit-learn#7525 from rickiepark/master

[MRG+2] add feature_name to diabetes dataset
weinberz · Oct 7, 2016 · 3d558fa · 3d558fa
2 parents 21a3c19 + 6c8a1b1
commit 3d558fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/sklearn/datasets/base.py b/sklearn/datasets/base.py
@@ -526,7 +526,9 @@ def load_diabetes(return_X_y=False):
     if return_X_y:
         return data, target
 
-    return Bunch(data=data, target=target)
+    return Bunch(data=data, target=target,
+                 feature_names=['age', 'sex', 'bmi', 'bp',
+                                's1', 's2', 's3', 's4', 's5', 's6'])
 
 
 def load_linnerud(return_X_y=False):

diff --git a/sklearn/datasets/tests/test_base.py b/sklearn/datasets/tests/test_base.py
@@ -171,6 +171,7 @@ def test_load_diabetes():
     res = load_diabetes()
     assert_equal(res.data.shape, (442, 10))
     assert_true(res.target.size, 442)
+    assert_equal(len(res.feature_names), 10)
 
     # test return_X_y option
     X_y_tuple = load_diabetes(return_X_y=True)