Only train on the first set of features because computation is otherwhise too long