This is wrong. You're checking the features of the first sample. (You're also not using broadcasting.)