from sklearn import datasets from sklearn.model_selection import train_test_split iris = datasets.load_iris() X = iris.data y = iris.target train_X, train_y, test_X, test_y = train_test_split(X, y, train_size=0.9) data_dict = { '_data': train_X, '_labels': train_y, '_test_data': test_X, '_test_labels': test_y } iris_data = Dataset(**data_dict)