machine-learning.functions: Add selu
parent
a515af7d68
commit
bfd830f6b4
|
@ -11,6 +11,10 @@ IN: machine-learning.functions
|
||||||
: leaky-relu ( x a -- x' )
|
: leaky-relu ( x a -- x' )
|
||||||
over 0 < [ * ] [ drop ] if ; inline
|
over 0 < [ * ] [ drop ] if ; inline
|
||||||
|
|
||||||
|
! https://arxiv.org/pdf/1706.02515.pdf
|
||||||
|
: selu ( x a -- x' )
|
||||||
|
over 0 < [ [ [ e^ ] dip * ] keep - ] [ drop ] if ; inline
|
||||||
|
|
||||||
: default-leaky-relu ( x -- x' )
|
: default-leaky-relu ( x -- x' )
|
||||||
.01 leaky-relu ; inline
|
.01 leaky-relu ; inline
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue