Add mean absolute error preset (#19)

reiinakano · Jun 3, 2017 · 335d10f · 335d10f
1 parent 258813c
commit 335d10f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/xcessiv/presets/metricsetting.py b/xcessiv/presets/metricsetting.py
@@ -11,6 +11,7 @@
     'precision_from_preds',
     'f1_score_from_scores',
     'f1_score_from_preds',
+    'mae',
     'mse',
     'roc_auc_score_from_scores'
 ]
@@ -196,6 +197,16 @@ def metric_generator(y_true, y_preds):
     'selection_name': 'F1 Score from Predictions'
 }
 
+mae = {
+    'name': 'Mean Absolute Error',
+    'source':
+    """from sklearn.metrics import mean_absolute_error
+
+metric_generator = mean_absolute_error
+""",
+    'selection_name': 'Mean Absolute Error'
+}
+
 mse = {
     'name': 'Mean Squared Error',
     'source':