demo/RR.R

In chenchunyu88/BATools: BATools: An R package that extends 'Bayesian Alphabet' for Whole Genome Prediction and Genome-wide association study

#This code demonstrate rrBLUP model
rm(list=ls())
library(BATools)
data("Pig")
#Standardize genotype matrix
geno=std_geno(PigM,method="s",freq=PigAlleleFreq)

init=set.init(~driploss,data=PigPheno,geno=geno,~id,df=5,pi_snp=1,h2=0.5,c=NULL,model="rrBLUP",centered=TRUE)
run_para=list(niter=2000,burnIn=1000,skip=10)
print_mcmc=list(piter=500)
update_para=list(df=FALSE,scale=TRUE,pi=FALSE)
op<-set.options(model="rrBLUP",method="MCMC",priors=NULL,init=init,
                   update_para=update_para,run_para=run_para,save.at="rrBLUP",print_mcmc=print_mcmc)

rrBLUP<-baFit(driploss~sex,data=PigPheno,geno=geno ,genoid = ~id,options = op)

#### Cross-validation using BATools
set.seed(1234)
PigPheno=createCV(~driploss,data = PigPheno,k=5)
cvrrBLUP<-baFit(driploss~sex,data=PigPheno,geno=geno ,genoid = ~id,options = op, train=~cv1)
baplot(cvrrBLUP)
par(mfrow=c(1,1))
cvrrBLUP