import numpy as np class Dataset(object): data = None target = None target_names = None def __repr__(self): rep = "Dataset(\n data: {0}, \n target: {1}, \n target_names: {2},\n)"\ .format(self.data, self.target, self.target_names) return rep def __init__(self, data=[], target=[], target_names=[]): self.data = np.array(data) self.target = np.array(target) self.target_names = np.array(target_names)