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

Learning_rate parameter in ContextualizedRegressionBase #50

Closed blengerich closed 2 years ago

blengerich commented 2 years ago

Set learning_rate in ContextualizedRegressionBase, so that PyTorch lightning helpers (auto_lr) can tune the hyperparameter.