class Data: """ This class defines the API for Data. """ def __init__(self, x, y): """ Constructor Parameters ---------- x : numpy array A numpy array of data samples y : numpy array A numpy array of outputs corresponding to the sample """ self.x = x self.y = y def __len__(self): """ Return the number of samples in the dataset Returns ------- int Number of samples """ return self.y.shape[0] def __getitem__(self, i): """ Function to get the item with index i. Parameters ---------- i : int Index Returns ------- 2-tuple A tuple of the ith data sample and it's corresponding label """ return self.x[i], self.y[i]