search

Aarhus-Psychiatry-Research / psycop-model-training

Shared code for model training and evaluation.
Other
1 stars 0 forks source link

Martbern/fix use agg backend to make matplotlib threadsafe #375

Closed MartinBernstorff closed 1 year ago

MartinBernstorff commented 1 year ago

Notes for reviewers

  1. Uses the agg backend to make matplotlib threadsafe
  2. Adds "self-healing" by returning an ROC-AUC of 0.5 to Optuna in case of an exception in the training loop. Handles if a config combination is impossible, e.g. no imputation, but using chi2. Logs the stacktrace to wandb, so should still be debuggable.