Skip to content

Commit 1174336

Browse files
Merge branch 'TGCN_non_linear' of https://github.com/juanmigutierrez/GraphNeuralNetworks.jl into TGCN_non_linear with fixes
2 parents b204a0a + ebee230 commit 1174336

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

GraphNeuralNetworks/src/layers/temporalconv.jl

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -867,9 +867,15 @@ Construct a recurrent layer corresponding to the [`TGCNCell`](@ref) cell.
867867
The arguments are passed to the [`TGCNCell`](@ref) constructor.
868868
See [`GNNRecurrence`](@ref) for more details.
869869
870+
# Additional Parameters
871+
872+
- `gate_activation`: Activation function for the gate mechanisms. Default `sigmoid`.
873+
870874
# Examples
871875
872876
```jldoctest
877+
julia> using Flux # Ensure relu is available
878+
873879
julia> num_nodes, num_edges = 5, 10;
874880
875881
julia> d_in, d_out = 2, 3;
@@ -896,5 +902,6 @@ julia> size(y) # (d_out, timesteps, num_nodes)
896902
(3, 5, 5)
897903
```
898904
"""
899-
TGCN(args...; kws...) = GNNRecurrence(TGCNCell(args...; kws...))
905+
TGCN(args...; act = relu, kws...) =
906+
GNNRecurrence(TGCNCell(args...; act, kws...))
900907

0 commit comments

Comments
 (0)