Machine learning predictions based on smaller sets of data and variables