Open nealgravindra opened 3 years ago
Need to review the classification scheme but this is on pause because we don't want to run hundreds of models before optimizing the main model architecture a bit more. As in, optimize for GA prediction, then apply to metadata variables
May be worthwhile to also try class imbalance adjustments to the loss function, such as: https://github.com/vandit15/Class-balanced-loss-pytorch/blob/master/class_balanced_loss.py
Cleaning up the figures:
Other enhancements:
ast.literal_eval