2

我有以下代码返回具有最低 AIC 的模型，但我希望所有模型的 AIC 以升序或降序排列，而不使用 R 中的内置排序函数

sp <- rnorm(100)  ## just some toy data to make code work!
spfinal.aic <- Inf
spfinal.order <- c(0,0,0)
for (p in 1:4) for (d in 0:1) for (q in 1:4) {
  spcurrent.aic <- AIC(arima(sp, order=c(p, d, q)))
   if (spcurrent.aic < spfinal.aic) {
     spfinal.aic <- spcurrent.aic
     spfinal.order <- c(p, d, q)
     spfinal.arima <- arima(sp, order=spfinal.order)
   }
}

我想spfinal.order<-c(p,d,p)成为 AIC 升序或降序的所有模型的列表。我怎样才能做到这一点？

4

1 回答 1

2

下面的代码可以满足您的需求。由于您想要记录所有已尝试的模型，因此不会在循环内进行比较。向量aic.vec将保存所有模型的 AIC 值，而矩阵order.matrix将逐列保存 ARIMA 规范。最后，我们按 AIC 值升序对两者进行排序，以便您知道最好的模型是第一个模型。

sp <- rnorm(100)  ## just some toy data to make code work!
order.matrix <- matrix(0, nrow = 3, ncol = 4 * 2 * 4)
aic.vec <- numeric(4 * 2 * 4)
k <- 1
for (p in 1:4) for (d in 0:1) for (q in 1:4) {
  order.matrix[, k] <- c(p,d,q)
  aic.vec[k] <- AIC(arima(sp, order=c(p, d, q)))
  k <- k+1
  }
ind <- order(aic.vec, decreasing = FALSE)
aic.vec <- aic.vec[ind]
order.matrix <- order.matrix[, ind]

我没有使用列表来存储 ARIMA 规范，因为我认为矩阵更好。目前矩阵是宽格式的，即3行多列。您可以转置它以获得更好的打印效果：

order.matrix <- t(order.matrix)

也许您还想绑定order.matrix在一起aic.vec以获得更好的演示？做这个：

result <- cbind(order.matrix, aic.vec)
colnames(result) <- c("p", "d", "q", "AIC")

我认为这使您更容易检查。示例输出（前 5 行）：

> result
      p d q      AIC
 [1,] 2 0 2 305.5698
 [2,] 3 0 3 305.8882
 [3,] 1 0 3 307.8365
 [4,] 2 1 3 307.9137
 [5,] 1 1 2 307.9952
于 2016-06-23T18:03:45.527 回答