I don't think these variable names are helpful. It makes it seems as if this is a subset of the training data, but it's a subspace of the features. Why not just do the slicing inline as before?