We then split 80% of the numbers into a training set and 20% into a test set.
we investigate this using a numerical extrapolation setting: we train models on a specific integer range and test them on values greater than the largest training number and smaller than the smallest training number.