some more progress. still got so much man... like you know, transfer learning, class imbalance, final results, early stopping