def get_data(): def f(x): result = numpy.zeros(3) if x == 'Iris-setosa': result[0] = 1 elif x == 'Iris-versicolor': result[1] = 1 elif x == 'Iris-virginica': result[2] = 1 else: raise ValueError return result data = File(u'../../../data') data.extract() data.shuffle() data.convertClass(function=f, length=3) return data.split(rate=0.5)