prediction with maxBy