search

comsaint / dlh_project

For CS598 Deep Learning for Healthcare project
0 stars 0 forks source link

Starter #12

Closed comsaint closed 3 years ago

comsaint commented 3 years ago
  1. Fix the plotting in evaluation method.
  2. Loss function now uses CrossEntropyLoss which can cater multiple label case if needed.
  3. Incorporate class weight to loss function to deal with class imbalance.