library(AIUQ)
msd_fn <- function(param, d_input){
beta = 2*param[1]^2
MSD = beta*d_input
}
msd_grad_fn <- function(param, d_input){
MSD_grad = 4*param[1]*d_input
}
theta = 2
d_input = 0:10
model_name = "user_defined"
MSD_list = get_MSD_with_grad(theta=theta,d_input=d_input,
model_name=model_name,msd_fn=msd_fn,
msd_grad_fn=msd_grad_fn)
MSD_list$msd
MSD_list$msd_grad
Run the code above in your browser using DataLab