tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

L2 Regularization

viveksembium opened this issue · comments

Is there an easy way to enable L2 Regularization to the skflow DNN classifier? Or is L2 enabled by default?

L2 regularization for learnable variables? Right now the easiest way would be to use custom model and add regularizer (from tensorflow) things - you can also add an option to TensorFlowDNN* estimator as a PR.

There's L2 regularization layer you can use in custom model from tensorflow.contrib.layers.