search

tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning
Apache License 2.0
3.18k stars 439 forks source link

L2 Regularization #139

Closed viveksembium closed 8 years ago

viveksembium commented 8 years ago

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

ilblackdragon commented 8 years ago

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.

terrytangyuan commented 8 years ago

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