jtdmCV: K-fold cross validation predictions and goodness of fit metrics
Description
Run K-fold cross validation predictions of the model m on a specified dataset.
Usage
jtdmCV(m, K = 5, sample = 1000, partition = NULL)
Value
A list containing:
Pred
Sample from the posterior predictive distribution in cross validation. It is an array where the first dimension is the number of sites in Xnew, the second is the number of traits modeled and the third the number of MCMC samples. NULL if FullPost=FALSE.
PredMean
Posterior mean of posterior predictive distribution in cross validation.
Predq975,Predq025
97.5% and 0.25% posterior quantiles of the posterior predictive distribution in cross validation. NULL if FullPost=FALSE.
R2
R squared of predictions in cross validation.
RMSE
Root square mean error between squared of predictions in cross validation.
Arguments
m
a model fitted with jtdm_fit
K
The number of folds of the K-fold cross validation
sample
Number of samples from the posterior distribution. Since we sample from the exact posterior distribution, the number of samples is relative lower than MCMC samplers. As a rule of thumb, 1000 samples should provide correct inference.
partition
A partition of the dataset specified by the user. It is a vector (whose length are the number of sites), where each element specifies the fold index of the site.