Easier early stopping in easy module

cnellington / Contextualized

An SKLearn-style toolbox for estimating and analyzing models, distributions, and functions with context-specific parameters.

http://contextualized.ml/

GNU General Public License v3.0

65 stars 9 forks source link

Easier early stopping in easy module #95

Closed blengerich closed 1 year ago

blengerich commented 2 years ago

Right now, we have to pass in callback_constructors kwarg to the SKLearnWrapper, but it may be more intuitive for users to just specify the patience instead.

cnellington commented 1 year ago

Related to #133 , has this been resolved @blengerich ?

blengerich commented 1 year ago

In dev but not main. Let's close when dev->main is pulled.

blengerich commented 1 year ago

Closing now because it's merged. Early stopping is controlled with es_monitor, es_mode, es_patience, es_min_delta, and es_verbose keyword arguments.