def __getitem__(self, i): X, y = self.X, self.y if is_pandas_ndframe(X): X = {k: X[k].values.reshape(-1, 1) for k in X} Xi = multi_indexing(X, i) yi = y if y is None else multi_indexing(y, i) return self.transform(Xi, yi)
def __getitem__(self, i): X, y = self.X, self.y if self.X_is_ndframe: X = {k: X[k].values.reshape(-1, 1) for k in X} Xi = multi_indexing(X, i, self.X_indexing) yi = multi_indexing(y, i, self.y_indexing) return self.transform(Xi, yi)