Packages
neural_net
1.0.0
NeuralNet is an A.I. library that allows for the construction and training of complex recurrent neural networks. Architectures such as LSTM or GRU can be specified in under 20 lines of code. Any neural network that can be built with the NeuralNet DSL can be trainined with automatically implemente...
Current section
Files
Jump to
Current section
Files
lib/rnn.ex
defmodule RNN do
@moduledoc "The classic simple recurrent neural network."
@doc "Takes an argument map with the keys :input_ids and :output_ids. Both values should be a list of component/id names for the input and output vectors (respectively)."
use NeuralNet
def template(inp, out \\ uid) do
tanh [inp, previous(out)], out
end
defp define(args) do
template(input, output)
def_vec(input, args.input_ids)
def_vec(output, args.output_ids)
end
end