From 3953a9a14fc441ca69c00024e3c6f6ab090f1bd4 Mon Sep 17 00:00:00 2001
From: rickiepark <haesunrpark@gmail.com>
Date: Thu, 29 Sep 2016 17:22:16 +0900
Subject: [PATCH] add feature_name to diabetes dataset (#4477)
---
sklearn/datasets/base.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/sklearn/datasets/base.py b/sklearn/datasets/base.py
index b5c7b4cd4f..924abc9615 100644
--- 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):
--
GitLab