Skip to content

Features: Metrics and CLI in inference.py file #51

Description

@seoulsky-field

What

  • Apply rich progress bar in inference.py file
  • Append more metrics in inference.py file

Why

In the train.py, we use rich to report in CLI, so I think it looks good to use rich in inference.py, too.
Also, AUROC score is used to metric score generally in medical task, however, we thought it's good to support more details and more metrics.

How

  • Apply rich progress bar in inference.py file.
  • Implement to get FPR, TPR, Best threshold.
  • Plot ROC curve with details.
  • Implement more metrics. (e.g. AUPRC, F1-score, Accuracy, etc.)

Metadata

Metadata

Labels

FeaturesNew feature or requestInferenceAbout benchmark inference fileRichAbout Rich (CLI Reporter)

Projects

Status
🏗 In progress

Relationships

None yet

Development

No branches or pull requests

Issue actions