Skip to content

New metrics for ensemble tuning objective #31

Description

@dylanbouchard

Is your feature request related to a problem? Please describe.
Currently, our UQEnsemble.tune and UQEnsemble.tune_from_graded methods offer the following metrics:
['fbeta_score', 'accuracy_score', 'balanced_accuracy_score', 'roc_auc', 'log_loss']. We we would like to include additional metrics, including, but not limited to, area under precision-recall curve and brier score.

Describe the solution you'd like
These metrics should be added to the uqlm.utils.Tuner class first, and offered by UQEnsemble methods by class composition (Tuner is stored as an attribute of UQEnsemble).

Describe alternatives you've considered
Current metrics

Additional context
Ideally, this issue should be worked on after #3 is completed and coded from scratch rather than relying on scikit-learn.

Metadata

Metadata

Assignees

Labels

Type

No type
No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions