SMART-Lab / smartlearner

SMART-Learner is a machine learning library built with researchers in mind.
BSD 3-Clause "New" or "Revised" License
10 stars 4 forks source link

Added EarlyStopping #29

Closed MarcCote closed 9 years ago

MarcCote commented 9 years ago

This PR also adds integration tests for stopping criteria: MaxEpochStopping and EarlyStopping.

MarcCote commented 9 years ago

@ASalvail at the end of early stopping the best model is restored (i.e. the shared variables contained in model.parameters).

I also added a Callback task. This is a recurrent task so we can call any Python function at the end of each K epoch or K update. On top of this, I created testing.py which will contain all the tools needed for doing tests.