Skip to content

Commit

Permalink
Update docs/src/full tutorials/Spam Detection with RNNs/SMS.jl
Browse files Browse the repository at this point in the history
Co-authored-by: Anthony Blaom, PhD <[email protected]>
  • Loading branch information
EssamWisam and ablaom authored Jun 7, 2024
1 parent 975ed6e commit e86009d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/src/full tutorials/Spam Detection with RNNs/SMS.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit e86009d

Please sign in to comment.