Simplified some of machine_learning_util.