search

sam-may / zephyr

Pneumonia segmentation model for CT scans. To be used in the context of covid-19 studies.
MIT License
0 stars 0 forks source link

Loss functions #15

Closed sam-may closed 4 years ago

sam-may commented 4 years ago

Now have ability to pass loss functions and associated hyperparameters as CLI. Should be very transparent to user with all the dirty work done in loss_functions.py.

Should be very transparent to user: the loss function CLI is a string which then gets converted into a function in loss_functions.py. A dictionary maps a given string to (1) a function and (2) a list of associated hyperparameters. The function choose_loss then converts this into an acceptable format to pass to model.compile.

sam-may commented 4 years ago

Going to wait for @jkguiang to merge the train-flow branch and then will restructure these changes on top of that.