From e86009d3dfbb480bbff1388c84eb7262970539ba Mon Sep 17 00:00:00 2001 From: Essam Date: Fri, 7 Jun 2024 17:12:05 +0300 Subject: [PATCH] Update docs/src/full tutorials/Spam Detection with RNNs/SMS.jl Co-authored-by: Anthony Blaom, PhD --- docs/src/full tutorials/Spam Detection with RNNs/SMS.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/src/full tutorials/Spam Detection with RNNs/SMS.jl b/docs/src/full tutorials/Spam Detection with RNNs/SMS.jl index 88692406..1570dc78 100644 --- a/docs/src/full tutorials/Spam Detection with RNNs/SMS.jl +++ b/docs/src/full tutorials/Spam Detection with RNNs/SMS.jl @@ -120,7 +120,7 @@ struct Mean end Flux.@layer Mean (m::Mean)(x) = mean(x, dims = 2)[:, 1, :] # [batch_size, seq_len, hidden_dim] => [batch_size, 1, hidden_dim]=> [batch_size, hidden_dim] -# For compatibility, we will also define a layer that simply casts the input to integers as the embedding layer in Flux expects integets but the MLJFlux model expects floats: +# For compatibility, we will also define a layer that simply casts the input to integers as the embedding layer in Flux expects integers but the MLJFlux model expects floats: struct Intify end Flux.@layer Intify (m::Intify)(x) = Int.(x)