diff --git a/README.rst b/README.rst index 515ade74..d0ebe814 100644 --- a/README.rst +++ b/README.rst @@ -65,7 +65,9 @@ like: objectives = ['accuracy'] [Output] - # again, these can/should be absolute paths + # Also compute the area under the ROC curve as an additional metric + metrics = ['roc_auc'] + # The following can/should be absolute paths log = output results = output predictions = output diff --git a/examples/titanic/evaluate_tuned.cfg b/examples/titanic/evaluate_tuned.cfg index 3c1bfc73..8f5bdd1d 100644 --- a/examples/titanic/evaluate_tuned.cfg +++ b/examples/titanic/evaluate_tuned.cfg @@ -18,7 +18,8 @@ objectives = ['accuracy'] [Output] # again, these can be absolute paths +metrics = ['roc_auc'] log = output results = output predictions = output -models = output \ No newline at end of file +models = output