rnn (version 0.8.1)

predict_gru: gru prediction function

Description

predict the output of a gru model

Usage

predict_gru(model, X, hidden = FALSE, real_output = T, ...)

Arguments

model

output of the trainr function

X

array of input values, dim 1: samples, dim 2: time, dim 3: variables (could be 1 or more, if a matrix, will be coerce to array)

hidden

should the function output the hidden units states

real_output

option used when the function in called inside trainr, do not drop factor for 2 dimension array output

...

arguments to pass on to sigmoid function

Value

array or matrix of predicted values