Advertisement
Guest User

Untitled

a guest
Jul 27th, 2017
55
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.47 KB | None | 0 0
  1. trait L2Regularization extends com.thoughtworks.deeplearning.plugins.INDArrayWeights {
  2. import org.nd4j.linalg.api.ndarray.INDArray
  3.  
  4. val l2Regularization: Double
  5.  
  6. trait INDArrayOptimizerApi extends super.INDArrayOptimizerApi {
  7. this: INDArrayOptimizer =>
  8. abstract override def delta: INDArray = {
  9. import org.nd4s.Implicits._
  10. super.delta + weight.data * l2Regularization
  11. }
  12. }
  13.  
  14. override type INDArrayOptimizer <: INDArrayOptimizerApi with Optimizer
  15. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement