Evaluate baseline bce model performance - #1
Conversation
Co-authored-by: denizcan.uelker <denizcan.uelker@mercedes-benz.com>
- Fixed missing num_items_in_batch parameter in CombinedLossTrainer.training_step() - Added emergency training scripts after BCE baseline failure (5.4% F1) - Created pragmatic training script for final model with Combined Loss (70% ASL) - Added monitoring scripts for training progress - Updated notebook with critical findings and new strategy
|
Cursor Agent can help with this pull request. Just |
There was a problem hiding this comment.
Hi @uelkerd! 👋
Your private repo does not have access to Sourcery.
Please upgrade to continue using Sourcery ✨
|
Important Review skippedDraft detected. Please check the settings in the CodeRabbit UI or the You can disable this status message by setting the Note 🎁 Summarized by CodeRabbit FreeYour organization is on the Free plan. CodeRabbit will generate a high-level summary and a walkthrough for each pull request. For a comprehensive line-by-line review, please upgrade your subscription to CodeRabbit Pro by visiting https://app.coderabbit.ai/login. 🪧 TipsChatThere are 3 ways to chat with CodeRabbit:
SupportNeed help? Join our Discord community for assistance with any issues or questions. CodeRabbit Commands (Invoked using PR/Issue comments)Type Other keywords and placeholders
CodeRabbit Configuration File (
|
Co-authored-by: denizcan.uelker <denizcan.uelker@mercedes-benz.com>
Co-authored-by: denizcan.uelker <denizcan.uelker@mercedes-benz.com>
Co-authored-by: denizcan.uelker <denizcan.uelker@mercedes-benz.com>
Co-authored-by: denizcan.uelker <denizcan.uelker@mercedes-benz.com>
Co-authored-by: denizcan.uelker <denizcan.uelker@mercedes-benz.com>
Fix
CombinedLossTrainerbug and implement Asymmetric/Combined Loss strategies with optimized hyperparameters to address catastrophic BCE baseline failure due to class imbalance.The BCE baseline model achieved an F1 Macro of only 0.054, learning only 3 out of 28 emotion classes, due to an extreme class imbalance ratio of 99.67. This PR introduces specialized loss functions (Asymmetric Loss, Combined Loss) and refined hyperparameters to effectively handle this imbalance and significantly improve model performance. A critical bug in
CombinedLossTrainerwas also fixed.