Add Early Stopping Mechanism

ravin-d-27 / PyDeepFlow

This is my own Deep Learning Package, optimized for performing Deep Learning Tasks and easy to learn and integrate into projects.

https://pypi.org/project/pydeepflow/

MIT License

18 stars 8 forks source link

Closed ravin-d-27 closed 1 month ago

ravin-d-27 commented 1 month ago

Implement early stopping to halt training when there is no significant improvement in validation loss, helping to prevent overfitting.

Objective:
Reduce unnecessary computation and improve model generalization.

Implementation Suggestions:

Create an EarlyStopping class that monitors validation loss.
Allow users to set a patience parameter, defining how many epochs to wait before stopping.
Provide feedback on whether early stopping was triggered during training.

pearguacamole commented 1 month ago

@ravin-d-27 Hey, Can you assign this to me, I'd like to work on this

ravin-d-27 commented 1 month ago

Hi @pearguacamole ! Thank you! Am assigning this issue to you.