Closed Oufattole closed 2 months ago
The updates introduce a new SupervisedModule
class for supervised learning tasks within a PyTorch Lightning framework. Additionally, a utils.py
file was added containing an OutputBase
dataclass for encapsulating loss attributes. These changes enhance the modularity and readability of the codebase by organizing supervised learning tasks and their outputs more effectively.
File(s) | Change Summary |
---|---|
src/meds_torch/model/supervised_model.py |
Introduced SupervisedModule class for supervised learning with methods for pretraining, finetuning, and metric tracking. |
src/meds_torch/model/utils.py |
Added OutputBase dataclass with a loss attribute for better output encapsulation. |
In the code where sparks of knowledge fly,
A SupervisedModule reaches for the sky.
Metrics and losses, all neatly in place,
WithOutputBase
, it keeps a steady pace.
Our code is a canvas, painted just right,
In the realm of AI, it's taking flight.
Thank you for using CodeRabbit. We offer it for free to the OSS community and would appreciate your support in helping us grow. If you find it useful, would you consider giving us a shout-out on your favorite social media?
Summary by CodeRabbit