From 9ae02dc65b1f560975632300ca23fcc02241c80b Mon Sep 17 00:00:00 2001
From: Olivier Grisel <olivier.grisel@ensta.org>
Date: Fri, 12 Nov 2010 04:46:37 +0100
Subject: [PATCH] make it possible to do fancy indexing on filenames
---
scikits/learn/datasets/base.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/scikits/learn/datasets/base.py b/scikits/learn/datasets/base.py
index 74570e2b8f..eee7420d89 100644
--- a/scikits/learn/datasets/base.py
+++ b/scikits/learn/datasets/base.py
@@ -96,7 +96,7 @@ def load_files(container_path, description=None, categories=None):
target.extend(len(documents) * [label])
filenames.extend(documents)
- return Bunch(filenames=filenames,
+ return Bunch(filenames=np.array(filenames),
target_names=target_names,
target=np.array(target),
DESCR=description)
--
GitLab