Skip to content
This repository was archived by the owner on Nov 22, 2022. It is now read-only.

Add field to specify model_select_metric for classification #222

Closed
wants to merge 1 commit into from

Conversation

mwu1993
Copy link
Contributor

@mwu1993 mwu1993 commented Jan 18, 2019

Summary: Currently, the ClassificationMetricReporter chooses the best model by accuracy on eval data (over the training epochs). Add a field to specify other metrics for selecting best model, namely roc_auc, mcc, macro_f1, f1 of a target label, avg precision of a target label, and loss.

Differential Revision: D13727079

@facebook-github-bot facebook-github-bot added the CLA Signed Do not delete this pull request or issue due to inactivity. label Jan 18, 2019
…research#222)

Summary:
Pull Request resolved: facebookresearch#222

Currently, the `ClassificationMetricReporter` chooses the best model by accuracy on eval data (over the training epochs). Add a field to specify other metrics for selecting best model, namely roc_auc, mcc, macro_f1, f1 of a target label, avg precision of a target label, and loss.

Reviewed By: rutyrinott

Differential Revision: D13727079

fbshipit-source-id: ce76d40511228d557ac2a24c842c9aaf1057f7ef
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed Do not delete this pull request or issue due to inactivity.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants