Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Enable custom loss, prediction targets, and reporting in scalar configs #442

Merged
merged 70 commits into from
Apr 21, 2021

Conversation

dccastro
Copy link
Member

Gives more flexibility for defining model configs with custom behaviour while leveraging the existing InnerEye workflows.

  • Add ScalarModelBase.target_names to define custom prediction targets (defaults to class_names).
  • Add ScalarModelBase.get_posthoc_label_transform() to post-process loaded labels for computing losses and metrics and generating reports.
  • Add ScalarLoss.CustomClassification/Regression and ScalarModelBase.get_loss_function().
  • Add ModelConfigBase.generate_custom_report()

dccastro and others added 30 commits February 23, 2021 12:06
- Metrics CSV files get loaded only once for each call to print_metrics()
- get_metric() now takes an optional threshold argument, so the computed
optimal threshold can be reused
@dccastro dccastro marked this pull request as ready for review April 21, 2021 09:58
CHANGELOG.md Outdated Show resolved Hide resolved
InnerEye/ML/scalar_config.py Outdated Show resolved Hide resolved
Shruthi42
Shruthi42 previously approved these changes Apr 21, 2021
@dccastro dccastro enabled auto-merge (squash) April 21, 2021 16:33
@dccastro dccastro merged commit 068704f into main Apr 21, 2021
@dccastro dccastro deleted the dacoelh/custom-labels branch April 21, 2021 17:28
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants