Factored out learning from regression_starter.py