Replace sampler balancing by loss balancing

This MR removes balancing of the training sampler and replaces it by loss balancing, if the loss function allows it (= contains a pos_weight parameter). Binary and exclusive, multi-class targets are supported. Using a database with multi-class, non-exclusive targets will raise a NotImplementedError for now.

Models now expect the class and arguments of the loss used instead of an instance, just like it is done with the optimizer. It is instantiated in trainer.py::run() and balanced if possible.

A base Model class has been created that other models should inherit from.

closes #6 (closed)

src/mednet/engine/trainer.py

@@ -93,8 +93,6 @@ def run(
         main_pid=os.getpid(),
     )
 
-    monitor_key = "loss/validation"
-
     # This checkpointer will operate at the end of every validation epoch
     # (which happens at each checkpoint period), it will then save the lowest
     # validation loss model observed.  It will also save the last trained model
@@ -102,7 +100,7 @@ def run(
         dirpath=output_folder,
         filename=CHECKPOINT_ALIASES["best"],
         save_last=True,  # will (re)create the last trained model, at every iteration
-        monitor=monitor_key,
+        monitor="loss/validation",
         mode="min",
         save_on_train_epoch_end=True,
         every_n_epochs=validation_period,  # frequency at which it checks the "monitor"