Allow custom activation functions

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

Closed blengerich closed 2 years ago

blengerich commented 2 years ago

ContextualRegressor can take in parameter activation to determine which torch.nn activation function to use. Default is nn.ReLU.