R/shin92protoalcove_opt.R

In ajwills72/catlearn: Formal Psychological Models of Categorization and Learning

## Run minimization
shin92protoalcove_opt <- function(params =  c(2,1,.25,.75),
                                  recompute = FALSE,
                                  trace = 0) {

    ##  Define objective function to minimize
    .alcove.shin92 <- function(params) {
        out <- shin92protoalcove(params)
        ret <- ssecl(out$c2p,catlearn::shin92$c2acc)
        ## 'shin92' for development 
        ## 'catlearn::shin92' for package release
        return(ret)
    }

    ## Set level of output verbosity
    ctrl <- list(trace=trace)
    
    ## Set bounds for optimzation
    lb <- c(0.0001,0.0001,0.0001,0.0001) # Lower bounds
    ub <- c(20,5,.99,.99)            # Upper bounds

    if(recompute) {
        ## Run optimzation
        names(params) <- c('c','phi','la','lw')
        result <- optim(params, .alcove.shin92, method = "L-BFGS-B",
                        lower = lb, upper = ub, control = ctrl)
        params <- result$par
    } else {
        ## Set result from stored copy
        params <- c(0.3617, 0.9161, 0.0196, 0.99)
    }
    
    return(params)
}