predict_lstm: gru prediction function

Description Usage Arguments Value

View source: R/predictr.R

Description

predict the output of a lstm model

Usage

1
predict_lstm(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 June 26, 2018, 5:02 p.m.