I think that you mean 'n_samples, n_features'. Let's stick to the same vocabulary as with the scikit-learn.