Skip to content

Threshold optimisation

Rohan Satya Isaac requested to merge risaac/lh-radiology-aimodel-service:thresholdOptimisation into master

Script to produce optimised thresholds for cheXnet model outputs.

  1. Loads the CXR8 dataset to a dataloader
  2. Use existing cheXnet model to produce predicted probability
  3. For each class, find best threshold by iterating through all cut offs. Use F1 score to finalize best threshold cutoff

Merge request reports