Fix bug in df_regr (#141)

Author: icfaust

sklearn_bench/df_regr.py

@@ -67,8 +67,8 @@ def main():
     parser = argparse.ArgumentParser(description='scikit-learn random forest '
                                      'regression benchmark')
 
-    parser.add_argument('--criterion', type=str, default='mse',
-                        choices=('mse', 'mae'),
+    parser.add_argument('--criterion', type=str, default='squared_error',
+                        choices=('squared_error', 'absolute_error'),
                         help='The function to measure the quality of a split')
     parser.add_argument('--num-trees', type=int, default=100,
                         help='Number of trees in the forest')