How do I create a sklearn.datasets.base.Bunch object in scikit-learn from my own data?

scikit-learn scikits
David picture David · Dec 10, 2013 · Viewed 17.2k times · Source

In most of the Scikit-learn algorithms, the data must be loaded as a Bunch object. For many example in the tutorial load_files() or other functions are used to populate the Bunch object. Functions like load_files() expect data to be present in certain format, but I have data stored in a different format, namely a CSV file with strings for each field.

How do I parse this and load data in the Bunch object format?

Answer

Hugh Perkins picture Hugh Perkins · Dec 21, 2016

You can do it like this:

import numpy as np
import sklearn.datasets

examples = []
examples.append('some text')
examples.append('another example text')
examples.append('example 3')

target = np.zeros((3,), dtype=np.int64)
target[0] = 0
target[1] = 1
target[2] = 0
dataset = sklearn.datasets.base.Bunch(data=examples, target=target)

Related questions

Cannot import scikits-learn even though it seems to be installed

Per the scikit-learn user guide, I installed scikit-learn using pip install -U scikit-learn. So using pip search scikit-learn, I get this search result: scikit-learn - A set of python modules for machine learning and data mining INSTALLED: 0.12.1 (latest) But when …

Read More
Missing values in scikits machine learning

Is it possible to have missing values in scikit-learn ? How should they be represented? I couldn't find any documentation about that.

Read More
Removing features with low variance using scikit-learn

scikit-learn provides various methods to remove descriptors, a basic method for this purpose has been provided by the given tutorial below, http://scikit-learn.org/stable/modules/feature_selection.html but the tutorial does not provide any method or a way …

Read More