R/sspFunctions.R
In JimGrange/flankr: Implements computational models of attentional selectivity
Defines functions
plotPredictionsSSP
predictionsSSP
fitMultipleSSP_fixed
fitSSP_fixed
fitMultipleSSP
fitSSP
simulateSSP
Documented in
fitMultipleSSP
fitMultipleSSP_fixed
fitSSP
fitSSP_fixed
simulateSSP
###
# Functions to run the SSP model itself. This includes functions to simulate
# data from the SSP model, as well as to run the fitting routine.
#------------------------------------------------------------------------------
#' Obtain simulated response times and accuracy from the SSP model
#'
#' \code{simulateSSP} generates synthetic data from the DSTP model in the
#' form of response time (RT) in seconds and accuracy for both congruent and
#' incongruent trials.
#'
#' This function can be employed by the user to generate synthetic data, but
#' its main purpose is to be used by the fitting procedure to generate model
#' predictions for a set of parameter values when trying to find the best-
#' fitting values.
#'
#' @param parms The set of parameters to use to simulate the data. Must be
#' contained in a vector in the order: \code{A}, \code{ter},
#' \code{p}, \code{rd}, \code{sda}.
#' @param nTrials How many trials to simulate per congruency condition.
#' @param var The variance of the diffusion process. By default this is set to
#' 0.01.
#' @param dt The diffusion scaling parameter (i.e., time steps). By default,
#' this is set to 0.001.
#' @param seed The value for the \code{set.seed} function to set random
#' generation state.
#'
#' @examples
#' # declare the parameters
#' parms <- c(0.050, 0.300, 0.400, 0.040, 1.500)
#'
#' # simulate the data
#' modelData <- simulateSSP(parms, nTrials = 10000)
#'
#' @return Returns a data frame with three columns: rt (response time) in
#' seconds, accuracy of the model's response (1 for correct, 0 for error), and
#' congruency condition.
#' @useDynLib flankr
#' @importFrom Rcpp sourceCpp
#' @export
simulateSSP <- function(parms, nTrials, var = 0.01, dt = 1/1000, seed = NULL){
# transfer nTrials to shorter name
n <- nTrials
# Set random number seed, so same predictions occur every time.
if(!is.null(seed)){
set.seed(seed, kind = NULL, normal.kind = NULL)
}
# initialise empty matrix for simulation data with two columns
# (RT & accuracy) and with rows = number of trials
trialData <- matrix(0, nrow = n * 2, ncol = 3)
colnames(trialData) <- c("rt", "accuracy", "congruency")
trialData <- data.frame(trialData)
# first generate congruent data by calling the C++ function
trialData[1:n, 1:2] <- getSSP(parms, trialType = 1, nTrials = n, dt, var)
trialData[1:n, 3] <- "congruent"
# now do incongruent data
trialData[(n + 1):(n * 2), 1:2] <- getSSP(parms, trialType = 2, nTrials = n,
dt, var)
trialData[(n + 1):(n * 2), 3] <- "incongruent"
return(trialData);
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
#' Fit the SSP model to human data
#'
#' \code{fitSSP} fits the SSP model to a single experimental condition of
#' human data (besides congruency, which it accounts for simutaneously).
#'
#' This function can be employed by the user to find the best-fitting
#' parameters of the SSP model to fit the human data of a single experimental
#' condition. The fitting procedure accounts for congruent and incongruent
#' trials simultaneously. The fit is obtained by a gradient-descent method
#' (using the Nelder-Mead method contained in R's \code{optim} function) and is
#' fit to the proportion of data contained in human CDF and CAF distributional
#' data.
#'
#' @param data A data frame containing human data. See \code{?exampleData} for
#' data formatted correctly.
#'
#' @param conditionName If there is an additional experimental manipulation
#' (i.e., other than target congruency) the model can only be fit to one at a
#' time. Tell the function which condition is currently being fit by passing
#' a string to the function (e.g., "present"). The function by default assumes
#' no additional condition (e.g., conditionName is set to NULL).
#'
#' @param parms A vector of starting parameters to use in the minimisation
#' routine. Must be in the order: \code{A}, \code{ter}, \code{p}, \code{rd},
#' \code{sda}.
#'
#' @param cdfs A vector of quantile values for cumulative distribution functions
#' to be estimated from the human data. The model will attempt to find the
#' best-fitting parameters that match this distributional data.
#'
#' @param cafs A vector of quantiles for conditional accuracy functions to be
#' estimated from the human data. The model will attempt to find the best-
#' fitting parameters that match this distributional data.
#'
#' @param maxParms A vector containing upper limits on possible parameter
#' values.
#'
#' @param nTrials An integer stating how many trials to simulate per iteration
#' of the fitting cycle for each congruency type.
#'
#' @param multipleSubjects A boolean stating whether the fit is to multiple
#' subjects (multipleSubjects = TRUE) or to a single subject
#' (multipleSubjects = FALSE).
#'
#' @return \code{bestParameters} A vector of the best-fitting parameters found
#' by the current fit run.
#'
#' @return \code{g2} The value of Wilks likelihood ratio (G2) obtained by the
#' current fit run.
#'
#' @return \code{bBIC} The value of the Bayesian Information Criterion (BIC)
#' obtained by the current fit run. This is calculated using the BIC equation
#' for binned data, hence bBIC (binned BIC).
#'
#' @examples
#' # Load the example data the comes with the \code{flankr} package
#' data(exampleData)
#'
#' # Fit the model to the condition "present" in the example data set using
#' # the default settings in the model.
#'
#' fit <- fitSSP(data = exampleData, conditionName = "present")
#'
#' # Fit the model using different CDF and CAF values, and 100,000 trials per
#' # fit cycle
#' cdfs <- c(.2, .4, .6, .8)
#' cafs <- c(.2, .4, .6, .8)
#'
#' fit <- fitSSP(exampleData, conditionName = "present", cdfs = cdfs,
#' cafs = cafs, nTrials = 100000)
#'
#'
#'@export
fitSSP<- function(data, conditionName = NULL,
parms = c(0.050, 0.300, 0.400, 0.050, 1.500),
cdfs = c(.1, .3, .5, .7, .9), cafs = c(.25, .50, .75),
maxParms = c(1, 1, 1, 1, 3), nTrials = 50000,
multipleSubjects = TRUE){
# declare the scaling on the parameters
parscale <- c(1, 10, 10, 1, 100)
# get the desired condition's data
if(is.null(conditionName)){
conditionData <- data
} else{
conditionData <- subset(data, data$condition == conditionName)
}
# get all of the distribution & proportion information from human data.
# This returns a list with all information in separate "cotainers" for ease
# of access & generalisation to different CDF and CAF sizes.
if(multipleSubjects == TRUE){
humanProportions <- getHumanProps(conditionData, cdfs, cafs)
} else {
humanProportions <- getHumanPropsSingle(conditionData, cdfs, cafs)
}
modelStart <- "Model Fit Running. Please Wait..."
print(modelStart)
# perform the fit
fit <- optim(parms, fn = fitFunctionSSP, humanProportions = humanProportions,
n = nTrials, maxParms = maxParms,
control = (parscale = parscale))
# what are the best-fitting parameters?
bestParameters <- round(fit$par, 3)
# what is the fit statistic value?
g2 <- fit$value
# get the approximate BIC value
bBIC <- bBIC(humanProportions, model = "SSP", parms = bestParameters,
nTrials = nTrials)
# put all results into a list, and return the list to the user
modelFit <- list(bestParameters = bestParameters, g2 = g2,
bBIC = bBIC)
modelFinished <- "Model Fit Finished."
print(modelFinished)
return(modelFit)
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
#' Fit the SSP model to human data with multiple starting parameters
#'
#' \code{fitSSPMultiple} fits the SSP model to a single experimental condition
#' of human data (besides congruency, which it accounts for simutaneously).
#'
#' This function can be employed by the user to find the best-fitting
#' parameters of the SSP model to fit the human data of a single experimental
#' condition. The fitting procedure accounts for congruent and incongruent
#' trials simultaneously. The fit is obtained by a gradient-descent method
#' (using the Nelder-Mead method contained in R's \code{optim} function) and is
#' fit to the proportion of data contained in human CDF and CAF distributional
#' data.
#'
#' @param data A data frame containing human data. See \code{?exampleData} for
#' data formatted correctly.
#'
#' @param conditionName If there is an additional experimental manipulation
#' (i.e., other than target congruency) the model can only be fit to one at a
#' time. Tell the function which condition is currently being fit by passing
#' a string to the function (e.g., "present"). The function by default assumes
#' no additional condition (e.g., conditionName is set to NULL).
#'
#' @param parms A vector of starting parameters to use in the minimisation
#' routine. Must be in the order: \code{A}, \code{ter}, \code{p}, \code{rd},
#' \code{sda}. These parameters will be the starting point for the random
#' parameters.
#'
#' @param var An integer stating the percentage of each parameter value that
#' should be used for finding random parameter starting points.
#'
#' @param nParms An integer stating how many random starting points to explore
#'
#' @param cdfs A vector of quantile values for cumulative distribution functions
#' to be estimated from the human data. The model will attempt to find the
#' best-fitting parameters that match this distributional data.
#'
#' @param cafs A vector of quantiles for conditional accuracy functions to be
#' estimated from the human data. The model will attempt to find the best-
#' fitting parameters that match this distributional data.
#'
#' @param maxParms A vector containing upper limits on possible parameter
#' values.
#'
#' @param nTrials An integer stating how many trials to simulate per iteration
#' of the fitting cycle for each congruency type.
#'
#' @param multipleSubjects A boolean stating whether the fit is to multiple
#' subjects (multipleSubjects = TRUE) or to a single subject
#' (multipleSubjects = FALSE).
#'
#' @return \code{bestParameters} A vector of the best-fitting parameters found
#' by the current fit run.
#'
#' @return \code{g2} The value of Wilks likelihood ratio (G2) obtained by the
#' current fit run.
#'
#' @return \code{bBIC} The value of the Bayesian Information Criterion (BIC)
#' obtained by the current fit run. This is calculated using the BIC equation
#' for binned data, hence bBIC (binned BIC).
#'
#' @examples
#' # Load the example data the comes with the \code{flankr} package
#' data(exampleData)
#'
#' # Fit the model to the condition "present" in the example data set using
#' # the default settings in the model.
#'
#' fit <- fitMultipleSSP(data = exampleData, conditionName = "present")
#'
#' # Fit the model using different CDF and CAF values, and 100,000 trials per
#' # fit cycle
#' cdfs <- c(.2, .4, .6, .8)
#' cafs <- c(.2, .4, .6, .8)
#'
#' fit <- fitMultipleSSP(exampleData, conditionName = "present", cdfs = cdfs,
#' cafs = cafs, nTrials = 100000)
#'
#'
#'@export
fitMultipleSSP <- function(data, conditionName = NULL,
parms = c(0.050, 0.300, 0.400, 0.050, 1.500),
var = 10, nParms = 20, cdfs = c(.1, .3, .5, .7, .9),
cafs = c(.25, .50, .75),
maxParms = c(1, 1, 1, 1, 2), nTrials = 50000,
multipleSubjects = TRUE){
# declare the scaling on the parameters
parscale <- c(1, 10, 10, 1, 100)
# get the desired condition's data
if(is.null(conditionName)){
conditionData <- data
} else{
conditionData <- subset(data, data$condition == conditionName)
}
# get all of the distribution & proportion information from human data.
# This returns a list with all information in separate "cotainers" for ease
# of access & generalisation to different CDF and CAF sizes.
if(multipleSubjects == TRUE){
humanProportions <- getHumanProps(conditionData, cdfs, cafs)
} else {
humanProportions <- getHumanPropsSingle(conditionData, cdfs, cafs)
}
# get random starting parameters
varParms <- (parms/ 100) * var
parameters <- getRandomParms(parms, varParms, maxParms, nParms)
#-------------
# Start the optimisation
modelStart <- "Model Fit Running. Please Wait..."
print(modelStart)
# initialise best-fitting parameters & best fit so far
bestFit <- .Machine$integer.max
bestParms <- numeric(length(parms))
bestBIC <- .Machine$integer.max
# start loop over all parameters now
for(i in 1:nParms){
# get the current run's parameters
currParms <- parameters[i, ]
fit <- optim(currParms, fn = fitFunctionSSP, humanProportions = humanProportions,
n = nTrials, maxParms = maxParms,
control = (parscale = parscale))
if(fit$value < bestFit){
bestFit <- fit$value
bestParms <- round(fit$par, 3)
bestBIC <- bBIC(humanProportions, model = "SSP", parms = bestParms,
nTrials = nTrials)
}
}
modelFit <- list(bestParameters = bestParms, g2 = bestFit,
bBIC = bestBIC)
modelFinished <- "Model Fit Finished."
print(modelFinished)
return(modelFit)
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
#' Fit the SSP model to human data with some fixed parameters
#'
#' \code{fitSSP_fixed} fits the SSP model to a single experimental condition of
#' human data (besides congruency, which it accounts for simutaneously). This
#' function allows the user to fix some parameters (using the \code{fixed}
#' variable).
#'
#' This function can be employed by the user to find the best-fitting
#' parameters of the SSP model to fit the human data of a single experimental
#' condition. The fitting procedure accounts for congruent and incongruent
#' trials simultaneously. The fit is obtained by a gradient-descent method
#' (using the Nelder-Mead method contained in R's \code{optim} function) and is
#' fit to the proportion of data contained in human CDF and CAF distributional
#' data.
#'
#' @param data A data frame containing human data. See \code{?exampleData} for
#' data formatted correctly.
#'
#' @param conditionName If there is an additional experimental manipulation
#' (i.e., other than target congruency) the model can only be fit to one at a
#' time. Tell the function which condition is currently being fit by passing
#' a string to the function (e.g., "present"). The function by default assumes
#' no additional condition (e.g., conditionName is set to NULL).
#'
#' @param parms A vector of starting parameters to use in the minimisation
#' routine. Must be in the order: \code{A}, \code{ter}, \code{p}, \code{rd},
#' \code{sda}.
#'
#' @param cdfs A vector of quantile values for cumulative distribution functions
#' to be estimated from the human data. The model will attempt to find the
#' best-fitting parameters that match this distributional data.
#'
#' @param cafs A vector of quantiles for conditional accuracy functions to be
#' estimated from the human data. The model will attempt to find the best-
#' fitting parameters that match this distributional data.
#'
#' @param maxParms A vector containing upper limits on possible parameter
#' values.
#'
#' @param nTrials An integer stating how many trials to simulate per iteration
#' of the fitting cycle for each congruency type.
#'
#' @param multipleSubjects A boolean stating whether the fit is to multiple
#' subjects (multipleSubjects = TRUE) or to a single subject
#' (multipleSubjects = FALSE).
#'
#' @param fixed A vector of TRUE/FALSE stating whether each parameter should be
#' fixed (TRUE) or free (FALSE) during the fitting routine. Must be in the
#' order: \code{A}, \code{ter}, \code{p}, \code{rd}, \code{sda}.
#'
#' @return \code{bestParameters} A vector of the best-fitting parameters found
#' by the current fit run.
#'
#' @return \code{g2} The value of Wilks likelihood ratio (G2) obtained by the
#' current fit run.
#'
#' @return \code{bBIC} The value of the Bayesian Information Criterion (BIC)
#' obtained by the current fit run. This is calculated using the BIC equation
#' for binned data, hence bBIC (binned BIC).
#'
#' @examples
#' # Load the example data the comes with the \code{flankr} package
#' data(exampleData)
#'
#' # Fit the model to the condition "present" in the example data set using
#' # the default settings in the model.
#'
#' fit <- fitSSP(data = exampleData, conditionName = "present")
#'
#' # Fix the first parameter (A) during the fit.
#'
#' parms = c(0.050, 0.300, 0.400, 0.050, 1.500)
#' fixed <- c(TRUE, FALSE, FALSE, FALSE, FALSE)
#' fit <- fitSSP_fixed(exampleData, conditionName = "present", parms = parms,
#' fixed = fixed)
#'
#'
#'@export
fitSSP_fixed <- function(data, conditionName = NULL,
parms = c(0.050, 0.300, 0.400, 0.050, 1.500),
cdfs = c(.1, .3, .5, .7, .9), cafs = c(.25, .50, .75),
maxParms = c(1, 1, 1, 1, 2), nTrials = 50000,
multipleSubjects = TRUE,
fixed = c(FALSE, FALSE, FALSE, FALSE, FALSE)){
# get the desired condition's data
if(is.null(conditionName)){
conditionData <- data
} else{
conditionData <- subset(data, data$condition == conditionName)
}
# get all of the distribution & proportion information from human data.
# This returns a list with all information in separate "cotainers" for ease
# of access & generalisation to different CDF and CAF sizes.
if(multipleSubjects == TRUE){
humanProportions <- getHumanProps(conditionData, cdfs, cafs)
} else {
humanProportions <- getHumanPropsSingle(conditionData, cdfs, cafs)
}
modelStart <- "Model Fit Running. Please Wait..."
print(modelStart)
# perform the fit using the wrapper function
fit <- optimFix_SSP(parms, fixed, humanProportions = humanProportions,
n = nTrials, maxParms = maxParms)
# what are the best-fitting parameters?
bestParameters <- round(fit$fullPars, 3)
# what is the fit statistic value?
g2 <- fit$value
# get the approximate BIC value
bBIC <- bBIC_fixed(humanProportions, model = "SSP", parms = bestParameters,
fixed = fixed, nTrials = nTrials)
# put all results into a list, and return the list to the user
modelFit <- list(bestParameters = bestParameters, g2 = g2,
bBIC = bBIC)
modelFinished <- "Model Fit Finished."
print(modelFinished)
return(modelFit)
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
#' Fit the SSP model to human data with mutiple starting parmaeters with some
#' fixed parameters
#'
#' \code{fitMultipleSSP_fixed} fits the SSP model to a single experimental
#' condition of human data (besides congruency, which it accounts for
#' simutaneously). This function explores multiple starting parameters and
#' allows user to fix model parameters.
#'
#' This function can be employed by the user to find the best-fitting
#' parameters of the SSP model to fit the human data of a single experimental
#' condition. The fitting procedure accounts for congruent and incongruent
#' trials simultaneously. The fit is obtained by a gradient-descent method
#' (using the Nelder-Mead method contained in R's \code{optim} function) and is
#' fit to the proportion of data contained in human CDF and CAF distributional
#' data. Multiple starting points of parameters are used.
#'
#' @param data A data frame containing human data. See \code{?exampleData} for
#' data formatted correctly.
#'
#' @param conditionName If there is an additional experimental manipulation
#' (i.e., other than target congruency) the model can only be fit to one at a
#' time. Tell the function which condition is currently being fit by passing
#' a string to the function (e.g., "present"). The function by default assumes
#' no additional condition (e.g., conditionName is set to NULL).
#'
#' @param parms A vector of starting parameters to use in the minimisation
#' routine. Must be in the order: \code{A}, \code{ter}, \code{p}, \code{rd},
#' \code{sda}. These parameters will be the starting point for the random
#' parameters.
#'
#' @param var An integer stating the percentage of each parameter value that
#' should be used for finding random parameter starting points.
#'
#' @param nParms An integer stating how many random starting points to explore
#'
#' @param cdfs A vector of quantile values for cumulative distribution functions
#' to be estimated from the human data. The model will attempt to find the
#' best-fitting parameters that match this distributional data.
#'
#' @param cafs A vector of quantiles for conditional accuracy functions to be
#' estimated from the human data. The model will attempt to find the best-
#' fitting parameters that match this distributional data.
#'
#' @param maxParms A vector containing upper limits on possible parameter
#' values.
#'
#' @param nTrials An integer stating how many trials to simulate per iteration
#' of the fitting cycle for each congruency type.
#'
#' @param multipleSubjects A boolean stating whether the fit is to multiple
#' subjects (multipleSubjects = TRUE) or to a single subject
#' (multipleSubjects = FALSE).
#'
#' @param fixed A vector of TRUE/FALSE stating whether each parameter should be
#' fixed (TRUE) or free (FALSE) during the fitting routine. Must be in the
#' order: \code{A}, \code{ter}, \code{p}, \code{rd}, \code{sda}.
#'
#' @return \code{bestParameters} A vector of the best-fitting parameters found
#' by the current fit run.
#'
#' @return \code{g2} The value of Wilks likelihood ratio (G2) obtained by the
#' current fit run.
#'
#' @return \code{bBIC} The value of the Bayesian Information Criterion (BIC)
#' obtained by the current fit run. This is calculated using the BIC equation
#' for binned data, hence bBIC (binned BIC).
#'
#' @examples
#' # Load the example data the comes with the \code{flankr} package
#' data(exampleData)
#'
#' # Fit the model to the condition "present" in the example data set using
#' # the default settings in the model.
#'
#' fit <- fitMultipleSSP(data = exampleData, conditionName = "present")
#'
#' # Fix the first parameter (A) during the fit.
#'
#' parms = c(0.050, 0.300, 0.400, 0.050, 1.500)
#' fixed <- c(TRUE, FALSE, FALSE, FALSE, FALSE)
#' fit <- fitMultipleSSP_fixed(exampleData, conditionName = "present",
#' parms = parms, fixed = fixed)
#'
#'@export
fitMultipleSSP_fixed <- function(data, conditionName = NULL,
parms = c(0.050, 0.300, 0.400, 0.050, 1.500),
var = 10, nParms = 20,
cdfs = c(.1, .3, .5, .7, .9),
cafs = c(.25, .50, .75),
maxParms = c(1, 1, 1, 1, 2),
nTrials = 50000, multipleSubjects = TRUE,
fixed = c(FALSE, FALSE, FALSE, FALSE, FALSE)){
# get the desired condition's data
if(is.null(conditionName)){
conditionData <- data
} else{
conditionData <- subset(data, data$condition == conditionName)
}
# get all of the distribution & proportion information from human data.
# This returns a list with all information in separate "cotainers" for ease
# of access & generalisation to different CDF and CAF sizes.
if(multipleSubjects == TRUE){
humanProportions <- getHumanProps(conditionData, cdfs, cafs)
} else {
humanProportions <- getHumanPropsSingle(conditionData, cdfs, cafs)
}
# get random starting parameters
varParms <- (parms/ 100) * var
parameters <- getRandomParms(parms, varParms, maxParms, nParms)
#-------------
# Start the optimisation
modelStart <- "Model Fit Running. Please Wait..."
print(modelStart)
# initialise best-fitting parameters & best fit so far
bestFit <- .Machine$integer.max
bestParms <- numeric(length(parms))
bestBIC <- .Machine$integer.max
# start loop over all parameters now
for(i in 1:nParms){
# get the current run's parameters
currParms <- parameters[i, ]
# check whether any parameter needs to be fixed
for(i in 1:length(currParms)){
if(fixed[i] == TRUE){
currParms[i] <- parms[i]
}
}
# run the fit whilst fixing necessary parameters
fit <- optimFix_SSP(parms, fixed, humanProportions = humanProportions,
n = nTrials, maxParms = maxParms)
if(fit$value < bestFit){
bestFit <- fit$value
bestParms <- round(fit$fullPars, 3)
bestBIC <- bBIC_fixed(humanProportions, model = "SSP", parms = bestParms,
fixed = fixed, nTrials = nTrials)
}
}
modelFit <- list(bestParameters = bestParms, g2 = bestFit,
bBIC = bestBIC)
modelFinished <- "Model Fit Finished."
print(modelFinished)
return(modelFit)
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
# Get the predicted proportions from the SSP model
#'@export
predictionsSSP<- function(parms, n, propsForModel, dt = 0.001, var = 0.01){
# parms = parameters for the model run
# n = number of trials per congruency condition
# propsForModel = CDF & CAF distributional information
# Run model to get congruent RTs
set.seed(42)
modelCon <- getSSP(parms, trialType = 1, n = n, dt, var)
modelConCDF <- getCDFProps(propsForModel$congruentCDFs, modelCon)
modelConCAF <- getCAFProps(propsForModel$congruentCAFsCutoff, modelCon)
set.seed(as.numeric(Sys.time()))
# Run model to get incontruent RTs
set.seed(42)
modelIncon <- getSSP(parms, trialType = 2, n = n, dt, var)
modelInconCDF <- getCDFProps(propsForModel$incongruentCDFs, modelIncon)
modelInconCAF <- getCAFProps(propsForModel$incongruentCAFsCutoff, modelIncon)
set.seed(as.numeric(Sys.time()))
modelProps <- list(modelCongruentCDF = modelConCDF,
modelCongruentCAF = modelConCAF,
modelIncongruentCDF = modelInconCDF,
modelIncongruentCAF = modelInconCAF)
return(modelProps)
}
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
# Get the predicted Quantiles from the DSTP model.
# This returns quantiles, not proportion per bin
# e.g., c(.1, .3, .5, .7, .9) not c(.1, .2, .2, .2, .2, 1)
#'@export
plotPredictionsSSP <- function(parms, n, propsForModel, dt = 0.001, var = 0.01){
# parms = parameters for the model run
# n = number of trials per congruency condition
# propsForModel = CDF & CAF distributional information
# Run model to get congruent RTs
set.seed(42)
modelCon <- getSSP(parms, trialType = 1, n = n, dt, var)
modelConCDF <- getModelCDFs(modelCon, propsForModel$congruentCDFs)
modelConCAF <- getModelCAFs(modelCon, propsForModel$congruentCAFsCutoff)
set.seed(as.numeric(Sys.time()))
# Run model to get incontruent RTs
set.seed(42)
modelIncon <- getSSP(parms, trialType = 2, n = n, dt, var)
modelInconCDF <- getModelCDFs(modelIncon, propsForModel$incongruentCDFs)
modelInconCAF <- getModelCAFs(modelIncon, propsForModel$incongruentCAFsCutoff)
set.seed(as.numeric(Sys.time()))
modelProps <- list(modelCongruentCDF = modelConCDF,
modelCongruentCAF = modelConCAF,
modelIncongruentCDF = modelInconCDF,
modelIncongruentCAF = modelInconCAF)
return(modelProps)
}
#------------------------------------------------------------------------------
JimGrange/flankr documentation built on Dec. 10, 2023, 12:17 a.m.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Defines functions plotPredictionsSSP predictionsSSP fitMultipleSSP_fixed fitSSP_fixed fitMultipleSSP fitSSP simulateSSP
Documented in fitMultipleSSP fitMultipleSSP_fixed fitSSP fitSSP_fixed simulateSSP
###
# Functions to run the SSP model itself. This includes functions to simulate
# data from the SSP model, as well as to run the fitting routine.
#------------------------------------------------------------------------------
#' Obtain simulated response times and accuracy from the SSP model
#'
#' \code{simulateSSP} generates synthetic data from the DSTP model in the
#' form of response time (RT) in seconds and accuracy for both congruent and
#' incongruent trials.
#'
#' This function can be employed by the user to generate synthetic data, but
#' its main purpose is to be used by the fitting procedure to generate model
#' predictions for a set of parameter values when trying to find the best-
#' fitting values.
#'
#' @param parms The set of parameters to use to simulate the data. Must be
#' contained in a vector in the order: \code{A}, \code{ter},
#' \code{p}, \code{rd}, \code{sda}.
#' @param nTrials How many trials to simulate per congruency condition.
#' @param var The variance of the diffusion process. By default this is set to
#' 0.01.
#' @param dt The diffusion scaling parameter (i.e., time steps). By default,
#' this is set to 0.001.
#' @param seed The value for the \code{set.seed} function to set random
#' generation state.
#'
#' @examples
#' # declare the parameters
#' parms <- c(0.050, 0.300, 0.400, 0.040, 1.500)
#'
#' # simulate the data
#' modelData <- simulateSSP(parms, nTrials = 10000)
#'
#' @return Returns a data frame with three columns: rt (response time) in
#' seconds, accuracy of the model's response (1 for correct, 0 for error), and
#' congruency condition.
#' @useDynLib flankr
#' @importFrom Rcpp sourceCpp
#' @export
simulateSSP <- function(parms, nTrials, var = 0.01, dt = 1/1000, seed = NULL){
# transfer nTrials to shorter name
n <- nTrials
# Set random number seed, so same predictions occur every time.
if(!is.null(seed)){
set.seed(seed, kind = NULL, normal.kind = NULL)
}
# initialise empty matrix for simulation data with two columns
# (RT & accuracy) and with rows = number of trials
trialData <- matrix(0, nrow = n * 2, ncol = 3)
colnames(trialData) <- c("rt", "accuracy", "congruency")
trialData <- data.frame(trialData)
# first generate congruent data by calling the C++ function
trialData[1:n, 1:2] <- getSSP(parms, trialType = 1, nTrials = n, dt, var)
trialData[1:n, 3] <- "congruent"
# now do incongruent data
trialData[(n + 1):(n * 2), 1:2] <- getSSP(parms, trialType = 2, nTrials = n,
dt, var)
trialData[(n + 1):(n * 2), 3] <- "incongruent"
return(trialData);
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
#' Fit the SSP model to human data
#'
#' \code{fitSSP} fits the SSP model to a single experimental condition of
#' human data (besides congruency, which it accounts for simutaneously).
#'
#' This function can be employed by the user to find the best-fitting
#' parameters of the SSP model to fit the human data of a single experimental
#' condition. The fitting procedure accounts for congruent and incongruent
#' trials simultaneously. The fit is obtained by a gradient-descent method
#' (using the Nelder-Mead method contained in R's \code{optim} function) and is
#' fit to the proportion of data contained in human CDF and CAF distributional
#' data.
#'
#' @param data A data frame containing human data. See \code{?exampleData} for
#' data formatted correctly.
#'
#' @param conditionName If there is an additional experimental manipulation
#' (i.e., other than target congruency) the model can only be fit to one at a
#' time. Tell the function which condition is currently being fit by passing
#' a string to the function (e.g., "present"). The function by default assumes
#' no additional condition (e.g., conditionName is set to NULL).
#'
#' @param parms A vector of starting parameters to use in the minimisation
#' routine. Must be in the order: \code{A}, \code{ter}, \code{p}, \code{rd},
#' \code{sda}.
#'
#' @param cdfs A vector of quantile values for cumulative distribution functions
#' to be estimated from the human data. The model will attempt to find the
#' best-fitting parameters that match this distributional data.
#'
#' @param cafs A vector of quantiles for conditional accuracy functions to be
#' estimated from the human data. The model will attempt to find the best-
#' fitting parameters that match this distributional data.
#'
#' @param maxParms A vector containing upper limits on possible parameter
#' values.
#'
#' @param nTrials An integer stating how many trials to simulate per iteration
#' of the fitting cycle for each congruency type.
#'
#' @param multipleSubjects A boolean stating whether the fit is to multiple
#' subjects (multipleSubjects = TRUE) or to a single subject
#' (multipleSubjects = FALSE).
#'
#' @return \code{bestParameters} A vector of the best-fitting parameters found
#' by the current fit run.
#'
#' @return \code{g2} The value of Wilks likelihood ratio (G2) obtained by the
#' current fit run.
#'
#' @return \code{bBIC} The value of the Bayesian Information Criterion (BIC)
#' obtained by the current fit run. This is calculated using the BIC equation
#' for binned data, hence bBIC (binned BIC).
#'
#' @examples
#' # Load the example data the comes with the \code{flankr} package
#' data(exampleData)
#'
#' # Fit the model to the condition "present" in the example data set using
#' # the default settings in the model.
#'
#' fit <- fitSSP(data = exampleData, conditionName = "present")
#'
#' # Fit the model using different CDF and CAF values, and 100,000 trials per
#' # fit cycle
#' cdfs <- c(.2, .4, .6, .8)
#' cafs <- c(.2, .4, .6, .8)
#'
#' fit <- fitSSP(exampleData, conditionName = "present", cdfs = cdfs,
#' cafs = cafs, nTrials = 100000)
#'
#'
#'@export
fitSSP<- function(data, conditionName = NULL,
parms = c(0.050, 0.300, 0.400, 0.050, 1.500),
cdfs = c(.1, .3, .5, .7, .9), cafs = c(.25, .50, .75),
maxParms = c(1, 1, 1, 1, 3), nTrials = 50000,
multipleSubjects = TRUE){
# declare the scaling on the parameters
parscale <- c(1, 10, 10, 1, 100)
# get the desired condition's data
if(is.null(conditionName)){
conditionData <- data
} else{
conditionData <- subset(data, data$condition == conditionName)
}
# get all of the distribution & proportion information from human data.
# This returns a list with all information in separate "cotainers" for ease
# of access & generalisation to different CDF and CAF sizes.
if(multipleSubjects == TRUE){
humanProportions <- getHumanProps(conditionData, cdfs, cafs)
} else {
humanProportions <- getHumanPropsSingle(conditionData, cdfs, cafs)
}
modelStart <- "Model Fit Running. Please Wait..."
print(modelStart)
# perform the fit
fit <- optim(parms, fn = fitFunctionSSP, humanProportions = humanProportions,
n = nTrials, maxParms = maxParms,
control = (parscale = parscale))
# what are the best-fitting parameters?
bestParameters <- round(fit$par, 3)
# what is the fit statistic value?
g2 <- fit$value
# get the approximate BIC value
bBIC <- bBIC(humanProportions, model = "SSP", parms = bestParameters,
nTrials = nTrials)
# put all results into a list, and return the list to the user
modelFit <- list(bestParameters = bestParameters, g2 = g2,
bBIC = bBIC)
modelFinished <- "Model Fit Finished."
print(modelFinished)
return(modelFit)
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
#' Fit the SSP model to human data with multiple starting parameters
#'
#' \code{fitSSPMultiple} fits the SSP model to a single experimental condition
#' of human data (besides congruency, which it accounts for simutaneously).
#'
#' This function can be employed by the user to find the best-fitting
#' parameters of the SSP model to fit the human data of a single experimental
#' condition. The fitting procedure accounts for congruent and incongruent
#' trials simultaneously. The fit is obtained by a gradient-descent method
#' (using the Nelder-Mead method contained in R's \code{optim} function) and is
#' fit to the proportion of data contained in human CDF and CAF distributional
#' data.
#'
#' @param data A data frame containing human data. See \code{?exampleData} for
#' data formatted correctly.
#'
#' @param conditionName If there is an additional experimental manipulation
#' (i.e., other than target congruency) the model can only be fit to one at a
#' time. Tell the function which condition is currently being fit by passing
#' a string to the function (e.g., "present"). The function by default assumes
#' no additional condition (e.g., conditionName is set to NULL).
#'
#' @param parms A vector of starting parameters to use in the minimisation
#' routine. Must be in the order: \code{A}, \code{ter}, \code{p}, \code{rd},
#' \code{sda}. These parameters will be the starting point for the random
#' parameters.
#'
#' @param var An integer stating the percentage of each parameter value that
#' should be used for finding random parameter starting points.
#'
#' @param nParms An integer stating how many random starting points to explore
#'
#' @param cdfs A vector of quantile values for cumulative distribution functions
#' to be estimated from the human data. The model will attempt to find the
#' best-fitting parameters that match this distributional data.
#'
#' @param cafs A vector of quantiles for conditional accuracy functions to be
#' estimated from the human data. The model will attempt to find the best-
#' fitting parameters that match this distributional data.
#'
#' @param maxParms A vector containing upper limits on possible parameter
#' values.
#'
#' @param nTrials An integer stating how many trials to simulate per iteration
#' of the fitting cycle for each congruency type.
#'
#' @param multipleSubjects A boolean stating whether the fit is to multiple
#' subjects (multipleSubjects = TRUE) or to a single subject
#' (multipleSubjects = FALSE).
#'
#' @return \code{bestParameters} A vector of the best-fitting parameters found
#' by the current fit run.
#'
#' @return \code{g2} The value of Wilks likelihood ratio (G2) obtained by the
#' current fit run.
#'
#' @return \code{bBIC} The value of the Bayesian Information Criterion (BIC)
#' obtained by the current fit run. This is calculated using the BIC equation
#' for binned data, hence bBIC (binned BIC).
#'
#' @examples
#' # Load the example data the comes with the \code{flankr} package
#' data(exampleData)
#'
#' # Fit the model to the condition "present" in the example data set using
#' # the default settings in the model.
#'
#' fit <- fitMultipleSSP(data = exampleData, conditionName = "present")
#'
#' # Fit the model using different CDF and CAF values, and 100,000 trials per
#' # fit cycle
#' cdfs <- c(.2, .4, .6, .8)
#' cafs <- c(.2, .4, .6, .8)
#'
#' fit <- fitMultipleSSP(exampleData, conditionName = "present", cdfs = cdfs,
#' cafs = cafs, nTrials = 100000)
#'
#'
#'@export
fitMultipleSSP <- function(data, conditionName = NULL,
parms = c(0.050, 0.300, 0.400, 0.050, 1.500),
var = 10, nParms = 20, cdfs = c(.1, .3, .5, .7, .9),
cafs = c(.25, .50, .75),
maxParms = c(1, 1, 1, 1, 2), nTrials = 50000,
multipleSubjects = TRUE){
# declare the scaling on the parameters
parscale <- c(1, 10, 10, 1, 100)
# get the desired condition's data
if(is.null(conditionName)){
conditionData <- data
} else{
conditionData <- subset(data, data$condition == conditionName)
}
# get all of the distribution & proportion information from human data.
# This returns a list with all information in separate "cotainers" for ease
# of access & generalisation to different CDF and CAF sizes.
if(multipleSubjects == TRUE){
humanProportions <- getHumanProps(conditionData, cdfs, cafs)
} else {
humanProportions <- getHumanPropsSingle(conditionData, cdfs, cafs)
}
# get random starting parameters
varParms <- (parms/ 100) * var
parameters <- getRandomParms(parms, varParms, maxParms, nParms)
#-------------
# Start the optimisation
modelStart <- "Model Fit Running. Please Wait..."
print(modelStart)
# initialise best-fitting parameters & best fit so far
bestFit <- .Machine$integer.max
bestParms <- numeric(length(parms))
bestBIC <- .Machine$integer.max
# start loop over all parameters now
for(i in 1:nParms){
# get the current run's parameters
currParms <- parameters[i, ]
fit <- optim(currParms, fn = fitFunctionSSP, humanProportions = humanProportions,
n = nTrials, maxParms = maxParms,
control = (parscale = parscale))
if(fit$value < bestFit){
bestFit <- fit$value
bestParms <- round(fit$par, 3)
bestBIC <- bBIC(humanProportions, model = "SSP", parms = bestParms,
nTrials = nTrials)
}
}
modelFit <- list(bestParameters = bestParms, g2 = bestFit,
bBIC = bestBIC)
modelFinished <- "Model Fit Finished."
print(modelFinished)
return(modelFit)
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
#' Fit the SSP model to human data with some fixed parameters
#'
#' \code{fitSSP_fixed} fits the SSP model to a single experimental condition of
#' human data (besides congruency, which it accounts for simutaneously). This
#' function allows the user to fix some parameters (using the \code{fixed}
#' variable).
#'
#' This function can be employed by the user to find the best-fitting
#' parameters of the SSP model to fit the human data of a single experimental
#' condition. The fitting procedure accounts for congruent and incongruent
#' trials simultaneously. The fit is obtained by a gradient-descent method
#' (using the Nelder-Mead method contained in R's \code{optim} function) and is
#' fit to the proportion of data contained in human CDF and CAF distributional
#' data.
#'
#' @param data A data frame containing human data. See \code{?exampleData} for
#' data formatted correctly.
#'
#' @param conditionName If there is an additional experimental manipulation
#' (i.e., other than target congruency) the model can only be fit to one at a
#' time. Tell the function which condition is currently being fit by passing
#' a string to the function (e.g., "present"). The function by default assumes
#' no additional condition (e.g., conditionName is set to NULL).
#'
#' @param parms A vector of starting parameters to use in the minimisation
#' routine. Must be in the order: \code{A}, \code{ter}, \code{p}, \code{rd},
#' \code{sda}.
#'
#' @param cdfs A vector of quantile values for cumulative distribution functions
#' to be estimated from the human data. The model will attempt to find the
#' best-fitting parameters that match this distributional data.
#'
#' @param cafs A vector of quantiles for conditional accuracy functions to be
#' estimated from the human data. The model will attempt to find the best-
#' fitting parameters that match this distributional data.
#'
#' @param maxParms A vector containing upper limits on possible parameter
#' values.
#'
#' @param nTrials An integer stating how many trials to simulate per iteration
#' of the fitting cycle for each congruency type.
#'
#' @param multipleSubjects A boolean stating whether the fit is to multiple
#' subjects (multipleSubjects = TRUE) or to a single subject
#' (multipleSubjects = FALSE).
#'
#' @param fixed A vector of TRUE/FALSE stating whether each parameter should be
#' fixed (TRUE) or free (FALSE) during the fitting routine. Must be in the
#' order: \code{A}, \code{ter}, \code{p}, \code{rd}, \code{sda}.
#'
#' @return \code{bestParameters} A vector of the best-fitting parameters found
#' by the current fit run.
#'
#' @return \code{g2} The value of Wilks likelihood ratio (G2) obtained by the
#' current fit run.
#'
#' @return \code{bBIC} The value of the Bayesian Information Criterion (BIC)
#' obtained by the current fit run. This is calculated using the BIC equation
#' for binned data, hence bBIC (binned BIC).
#'
#' @examples
#' # Load the example data the comes with the \code{flankr} package
#' data(exampleData)
#'
#' # Fit the model to the condition "present" in the example data set using
#' # the default settings in the model.
#'
#' fit <- fitSSP(data = exampleData, conditionName = "present")
#'
#' # Fix the first parameter (A) during the fit.
#'
#' parms = c(0.050, 0.300, 0.400, 0.050, 1.500)
#' fixed <- c(TRUE, FALSE, FALSE, FALSE, FALSE)
#' fit <- fitSSP_fixed(exampleData, conditionName = "present", parms = parms,
#' fixed = fixed)
#'
#'
#'@export
fitSSP_fixed <- function(data, conditionName = NULL,
parms = c(0.050, 0.300, 0.400, 0.050, 1.500),
cdfs = c(.1, .3, .5, .7, .9), cafs = c(.25, .50, .75),
maxParms = c(1, 1, 1, 1, 2), nTrials = 50000,
multipleSubjects = TRUE,
fixed = c(FALSE, FALSE, FALSE, FALSE, FALSE)){
# get the desired condition's data
if(is.null(conditionName)){
conditionData <- data
} else{
conditionData <- subset(data, data$condition == conditionName)
}
# get all of the distribution & proportion information from human data.
# This returns a list with all information in separate "cotainers" for ease
# of access & generalisation to different CDF and CAF sizes.
if(multipleSubjects == TRUE){
humanProportions <- getHumanProps(conditionData, cdfs, cafs)
} else {
humanProportions <- getHumanPropsSingle(conditionData, cdfs, cafs)
}
modelStart <- "Model Fit Running. Please Wait..."
print(modelStart)
# perform the fit using the wrapper function
fit <- optimFix_SSP(parms, fixed, humanProportions = humanProportions,
n = nTrials, maxParms = maxParms)
# what are the best-fitting parameters?
bestParameters <- round(fit$fullPars, 3)
# what is the fit statistic value?
g2 <- fit$value
# get the approximate BIC value
bBIC <- bBIC_fixed(humanProportions, model = "SSP", parms = bestParameters,
fixed = fixed, nTrials = nTrials)
# put all results into a list, and return the list to the user
modelFit <- list(bestParameters = bestParameters, g2 = g2,
bBIC = bBIC)
modelFinished <- "Model Fit Finished."
print(modelFinished)
return(modelFit)
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
#' Fit the SSP model to human data with mutiple starting parmaeters with some
#' fixed parameters
#'
#' \code{fitMultipleSSP_fixed} fits the SSP model to a single experimental
#' condition of human data (besides congruency, which it accounts for
#' simutaneously). This function explores multiple starting parameters and
#' allows user to fix model parameters.
#'
#' This function can be employed by the user to find the best-fitting
#' parameters of the SSP model to fit the human data of a single experimental
#' condition. The fitting procedure accounts for congruent and incongruent
#' trials simultaneously. The fit is obtained by a gradient-descent method
#' (using the Nelder-Mead method contained in R's \code{optim} function) and is
#' fit to the proportion of data contained in human CDF and CAF distributional
#' data. Multiple starting points of parameters are used.
#'
#' @param data A data frame containing human data. See \code{?exampleData} for
#' data formatted correctly.
#'
#' @param conditionName If there is an additional experimental manipulation
#' (i.e., other than target congruency) the model can only be fit to one at a
#' time. Tell the function which condition is currently being fit by passing
#' a string to the function (e.g., "present"). The function by default assumes
#' no additional condition (e.g., conditionName is set to NULL).
#'
#' @param parms A vector of starting parameters to use in the minimisation
#' routine. Must be in the order: \code{A}, \code{ter}, \code{p}, \code{rd},
#' \code{sda}. These parameters will be the starting point for the random
#' parameters.
#'
#' @param var An integer stating the percentage of each parameter value that
#' should be used for finding random parameter starting points.
#'
#' @param nParms An integer stating how many random starting points to explore
#'
#' @param cdfs A vector of quantile values for cumulative distribution functions
#' to be estimated from the human data. The model will attempt to find the
#' best-fitting parameters that match this distributional data.
#'
#' @param cafs A vector of quantiles for conditional accuracy functions to be
#' estimated from the human data. The model will attempt to find the best-
#' fitting parameters that match this distributional data.
#'
#' @param maxParms A vector containing upper limits on possible parameter
#' values.
#'
#' @param nTrials An integer stating how many trials to simulate per iteration
#' of the fitting cycle for each congruency type.
#'
#' @param multipleSubjects A boolean stating whether the fit is to multiple
#' subjects (multipleSubjects = TRUE) or to a single subject
#' (multipleSubjects = FALSE).
#'
#' @param fixed A vector of TRUE/FALSE stating whether each parameter should be
#' fixed (TRUE) or free (FALSE) during the fitting routine. Must be in the
#' order: \code{A}, \code{ter}, \code{p}, \code{rd}, \code{sda}.
#'
#' @return \code{bestParameters} A vector of the best-fitting parameters found
#' by the current fit run.
#'
#' @return \code{g2} The value of Wilks likelihood ratio (G2) obtained by the
#' current fit run.
#'
#' @return \code{bBIC} The value of the Bayesian Information Criterion (BIC)
#' obtained by the current fit run. This is calculated using the BIC equation
#' for binned data, hence bBIC (binned BIC).
#'
#' @examples
#' # Load the example data the comes with the \code{flankr} package
#' data(exampleData)
#'
#' # Fit the model to the condition "present" in the example data set using
#' # the default settings in the model.
#'
#' fit <- fitMultipleSSP(data = exampleData, conditionName = "present")
#'
#' # Fix the first parameter (A) during the fit.
#'
#' parms = c(0.050, 0.300, 0.400, 0.050, 1.500)
#' fixed <- c(TRUE, FALSE, FALSE, FALSE, FALSE)
#' fit <- fitMultipleSSP_fixed(exampleData, conditionName = "present",
#' parms = parms, fixed = fixed)
#'
#'@export
fitMultipleSSP_fixed <- function(data, conditionName = NULL,
parms = c(0.050, 0.300, 0.400, 0.050, 1.500),
var = 10, nParms = 20,
cdfs = c(.1, .3, .5, .7, .9),
cafs = c(.25, .50, .75),
maxParms = c(1, 1, 1, 1, 2),
nTrials = 50000, multipleSubjects = TRUE,
fixed = c(FALSE, FALSE, FALSE, FALSE, FALSE)){
# get the desired condition's data
if(is.null(conditionName)){
conditionData <- data
} else{
conditionData <- subset(data, data$condition == conditionName)
}
# get all of the distribution & proportion information from human data.
# This returns a list with all information in separate "cotainers" for ease
# of access & generalisation to different CDF and CAF sizes.
if(multipleSubjects == TRUE){
humanProportions <- getHumanProps(conditionData, cdfs, cafs)
} else {
humanProportions <- getHumanPropsSingle(conditionData, cdfs, cafs)
}
# get random starting parameters
varParms <- (parms/ 100) * var
parameters <- getRandomParms(parms, varParms, maxParms, nParms)
#-------------
# Start the optimisation
modelStart <- "Model Fit Running. Please Wait..."
print(modelStart)
# initialise best-fitting parameters & best fit so far
bestFit <- .Machine$integer.max
bestParms <- numeric(length(parms))
bestBIC <- .Machine$integer.max
# start loop over all parameters now
for(i in 1:nParms){
# get the current run's parameters
currParms <- parameters[i, ]
# check whether any parameter needs to be fixed
for(i in 1:length(currParms)){
if(fixed[i] == TRUE){
currParms[i] <- parms[i]
}
}
# run the fit whilst fixing necessary parameters
fit <- optimFix_SSP(parms, fixed, humanProportions = humanProportions,
n = nTrials, maxParms = maxParms)
if(fit$value < bestFit){
bestFit <- fit$value
bestParms <- round(fit$fullPars, 3)
bestBIC <- bBIC_fixed(humanProportions, model = "SSP", parms = bestParms,
fixed = fixed, nTrials = nTrials)
}
}
modelFit <- list(bestParameters = bestParms, g2 = bestFit,
bBIC = bestBIC)
modelFinished <- "Model Fit Finished."
print(modelFinished)
return(modelFit)
} # end of function
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
# Get the predicted proportions from the SSP model
#'@export
predictionsSSP<- function(parms, n, propsForModel, dt = 0.001, var = 0.01){
# parms = parameters for the model run
# n = number of trials per congruency condition
# propsForModel = CDF & CAF distributional information
# Run model to get congruent RTs
set.seed(42)
modelCon <- getSSP(parms, trialType = 1, n = n, dt, var)
modelConCDF <- getCDFProps(propsForModel$congruentCDFs, modelCon)
modelConCAF <- getCAFProps(propsForModel$congruentCAFsCutoff, modelCon)
set.seed(as.numeric(Sys.time()))
# Run model to get incontruent RTs
set.seed(42)
modelIncon <- getSSP(parms, trialType = 2, n = n, dt, var)
modelInconCDF <- getCDFProps(propsForModel$incongruentCDFs, modelIncon)
modelInconCAF <- getCAFProps(propsForModel$incongruentCAFsCutoff, modelIncon)
set.seed(as.numeric(Sys.time()))
modelProps <- list(modelCongruentCDF = modelConCDF,
modelCongruentCAF = modelConCAF,
modelIncongruentCDF = modelInconCDF,
modelIncongruentCAF = modelInconCAF)
return(modelProps)
}
#------------------------------------------------------------------------------
#------------------------------------------------------------------------------
# Get the predicted Quantiles from the DSTP model.
# This returns quantiles, not proportion per bin
# e.g., c(.1, .3, .5, .7, .9) not c(.1, .2, .2, .2, .2, 1)
#'@export
plotPredictionsSSP <- function(parms, n, propsForModel, dt = 0.001, var = 0.01){
# parms = parameters for the model run
# n = number of trials per congruency condition
# propsForModel = CDF & CAF distributional information
# Run model to get congruent RTs
set.seed(42)
modelCon <- getSSP(parms, trialType = 1, n = n, dt, var)
modelConCDF <- getModelCDFs(modelCon, propsForModel$congruentCDFs)
modelConCAF <- getModelCAFs(modelCon, propsForModel$congruentCAFsCutoff)
set.seed(as.numeric(Sys.time()))
# Run model to get incontruent RTs
set.seed(42)
modelIncon <- getSSP(parms, trialType = 2, n = n, dt, var)
modelInconCDF <- getModelCDFs(modelIncon, propsForModel$incongruentCDFs)
modelInconCAF <- getModelCAFs(modelIncon, propsForModel$incongruentCAFsCutoff)
set.seed(as.numeric(Sys.time()))
modelProps <- list(modelCongruentCDF = modelConCDF,
modelCongruentCAF = modelConCAF,
modelIncongruentCDF = modelInconCDF,
modelIncongruentCAF = modelInconCAF)
return(modelProps)
}
#------------------------------------------------------------------------------
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Embedding an R snippet on your website
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.