demo/broydt_test.R

rm(list = ls())
library(snewton)

broydt.f <- function(x) {
    n <- length(x)
    res <- rep(NA, n)
    res[1] <- ((3 - 0.5 * x[1]) * x[1]) - 2 * x[2] + 1
    tnm1 <- 2:(n - 1)
    res[tnm1] <- ((3 - 0.5 * x[tnm1]) * x[tnm1]) - x[tnm1 - 1] - 
        2 * x[tnm1 + 1] + 1
    res[n] <- ((3 - 0.5 * x[n]) * x[n]) - x[n - 1] + 1
    sum(res * res)
}

broydt.g <- function(x) {
    n <- length(x)
    gg <- rep(NA, n)  # gradient set to NA to start with
    gg[1] <- -2 + 2 * x[1] + 4 * x[3] + (6 - 2 * x[1]) * (1 - 
        2 * x[2] + x[1] * (3 - 0.5 * x[1])) - 2 * x[2] * (3 - 
        0.5 * x[2])
    gg[2] <- -6 + 4 * x[4] + 10 * x[2] + (6 - 2 * x[2]) * (1 - 
        x[1] - 2 * x[3] + x[2] * (3 - 0.5 * x[2])) - 4 * x[1] * 
        (3 - 0.5 * x[1]) - 2 * x[3] * (3 - 0.5 * x[3])
    tnm2 <- 3:(n - 2)
    gg[tnm2] <- -6 + 4 * x[tnm2 - 2] + 4 * x[tnm2 + 2] + 10 * 
        x[tnm2] + (6 - 2 * x[tnm2]) * (1 - x[tnm2 - 1] - 2 * 
        x[tnm2 + 1] + x[tnm2] * (3 - 0.5 * x[tnm2])) - 4 * x[tnm2 - 
        1] * (3 - 0.5 * x[tnm2 - 1]) - 2 * x[tnm2 + 1] * (3 - 
        0.5 * x[tnm2 + 1])
    gg[n - 1] <- -6 + 4 * x[n - 3] + 10 * x[n - 1] + (6 - 2 * 
        x[n - 1]) * (1 - x[n - 2] - 2 * x[n] + x[n - 1] * (3 - 
        0.5 * x[n - 1])) - 4 * x[n - 2] * (3 - 0.5 * x[n - 2]) - 
        2 * x[n] * (3 - 0.5 * x[n])
    
    gg[n] <- -4 + 4 * x[n - 2] + 8 * x[n] + (6 - 2 * x[n]) * 
        (1 - x[n - 1] + x[n] * (3 - 0.5 * x[n])) - 4 * x[n - 
        1] * (3 - 0.5 * x[n - 1])
    return(gg)
}

##?? Need hessian

ni <- c(1, 2, 3)

times <- matrix(NA, nrow = 3, ncol = 4)


for (ii in ni) {
    n <- 4^ii  # Change from 10^ii 20100525 to reduce testing time
    cat("n=", n, "\n")
#    afname <- paste("ansbroydt", n, "UG", sep = "")
    x0 <- rep(pi, n)
    ut <- system.time(ans <- snewton(x0, broydt.f, broydt.g, control = list(trace = 1)))[1]
    times[ii, 1] <- ut
    cat("unconstrained with gradient\n")
#    sink(afname, split=TRUE)
    print(ans)
#    sink()
#    afname <- paste("ansbroydt", n, "UN", sep = "")
    ut <- system.time(ans <- snewton(x0, broydt.f, control = list(trace = 1)))[1]
    times[ii, 2] <- ut
    cat("unconstrained, no analytic gradient\n")
#    sink(afname, split=TRUE)
    print(ans)
#    sink()
    
    lower <- rep(1, n)
    upper <- rep(Inf, n)
    
#    afname <- paste("ansbroydt", n, "BG", sep = "")
    x0 <- rep(pi, n)
    ut <- system.time(ans <- snewton(x0, broydt.f, broydt.g, lower = lower, 
        upper = upper, control = list(trace = 1)))[1]
    times[ii, 3] <- ut
    cat("constrained, no analytic gradient\n")
#    sink(afname, split=TRUE)
    print(ans)
#    sink()
#    afname <- paste("ansbroydt", n, "BN", sep = "")
    ut <- system.time(ans <- snewton(x0, broydt.f, lower = lower, 
        upper = upper, control = list(trace = 1)))[1]
    times[ii, 4] <- ut
    cat("constrained, no analytic gradient\n")
#    sink(afname, split=TRUE)
    print(ans)
#    sink()
}

cat("Cols are UG, UN, BG, BN times\n")
print(times)

Try the snewton package in your browser

Any scripts or data that you put into this service are public.

snewton documentation built on May 2, 2019, 6:47 p.m.