predict_gru: gru prediction function

In rnn: Recurrent Neural Network

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

rnn documentation built on April 22, 2023, 1:12 a.m.