ts_lstm {daltoolbox}

LSTM

Description

Creates a time series prediction object that uses the LSTM. It wraps the pytorch library.

Usage

ts_lstm(preprocess = NA, input_size = NA, epochs = 10000L)

Arguments

preprocess

normalization

input_size

input size for machine learning model

epochs

maximum number of epochs

Value

a ts_lstm object.

Examples

#Use the same example of ts_mlp changing the constructor to:
model <- ts_lstm(ts_norm_gminmax(), input_size=4, epochs = 10000L)