train.ts: Train function to support time series objects

Description Usage Arguments Value Examples

View source: R/train.R

Description

Extends the train function from the "caret" package to support time series objects.

Usage

1
2
## S3 method for class 'ts'
train(x, ...)

Arguments

x

Object of class ts.

...

Further arguments passed on to train for training.

Value

Trained model.

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
library(caret)

library(forecast)
data(WWWusage)

class(WWWusage)
str(WWWusage)

arima <- train(WWWusage, method = auto_arima_model(), trControl = trainDirectFit())
summary(arima)
arimaorder(arima$finalModel) # order of best model
RMSE(predict(arima, WWWusage), WWWusage) # in-sample RMSE

library(vars)
data(Canada)

class(Canada)
str(Canada)

arima <- train(x = Canada[, -2], y = Canada[, 2], 
               method = auto_arima_model(), trControl = trainDirectFit())
summary(arima)
arimaorder(arima$finalModel) # order of best model
RMSE(predict(arima, Canada[, -2]), Canada[, 2]) # in-sample RMSE

arima <- train(form = prod ~ ., data = Canada, 
               method = auto_arima_model(), trControl = trainDirectFit())
summary(arima)
arimaorder(arima$finalModel) # order of best model
RMSE(predict(arima, Canada), Canada[, "prod"]) # in-sample RMSE

sfeuerriegel/caret.ts documentation built on May 29, 2019, 8:01 p.m.