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)