inst/scripts/create.trainset.R

In gbonte/D2C: Predicting Causal Direction from Dependency Features

rm(list=ls())

require(gbcode)
library(D2C)
library(doParallel)
type="is.parent"

is.what<-function(iDAG,i,j){
  if (type=="is.mb")
    return(as.numeric(is.mb(iDAG,i,j)))
  if (type=="is.parent")
    return(as.numeric(is.parent(iDAG,i,j)))
  
  if (type=="is.child")
    return(as.numeric(is.child(iDAG,i,j)))
  if (type=="is.descendant")
    return(as.numeric(is.descendant(iDAG,i,j)))
  
  if (type=="is.ancestor")
    return(as.numeric(is.ancestor(iDAG,i,j)))
  
}

set.seed(0)
noNodes<-20

N<-150 #c(150,200)
## range of number of samples

sdev<-c(0.1,0.5)
goParallel=FALSE

nfeat=20
maxs=10
ncores=1


NDAG=100 ## total number of DAGS
dN=5   ## number of DAGs per iteration
if (goParallel){
  ncores=20
  cl <- makeForkCluster(ncores)
  registerDoParallel(cl)
  dN=3*ncores
}

  

iter=1
rep=1
while ( iter <=NDAG){
  
  set.seed(iter)

  trainDAG<-new("simulatedDAG",NDAG=dN, N=N, noNodes=noNodes,
              functionType = c("linear","quadratic","sigmoid"),
              seed=iter,sdn=sdev,verbose=TRUE,
              additive=c(TRUE,FALSE),goParallel=goParallel)
  cat("Computed trainDAG \n")
  
  
  descr<-new("D2C.descriptor",bivariate=TRUE,ns=8,maxs=maxs,
             acc=TRUE,diff=TRUE,resid=TRUE,
             lin=TRUE,struct=TRUE, boot="mimr",pq=c(0.1,0.5,0.9))

  
  D2C<-new("D2C",sDAG=trainDAG,
           descr=descr,ratioEdges=0.2,
           max.features=nfeat, type=type,goParallel=goParallel,
           verbose=TRUE,npar=min(NDAG,ncores))

   

  if (iter==1){
    allD2C<-D2C
  } else{
    allD2C@origX<-rbind(allD2C@origX,D2C@origX)
    allD2C@Y<-c(allD2C@Y,D2C@Y)
    cat("dim(allD2C@origX)=",dim(allD2C@origX),"NDAG computed=",iter+dN,"\n")
  }

  iter=iter+dN

    namefile<-paste("./data/trainD2Cp",NDAG,max(noNodes),type,"RData",sep=".")
    trainD2C<-makeModel(allD2C,classifier="RF",EErep=2)
    save(file=namefile,list=c("trainD2C","allD2C","descr"))
    cat("SAVED ", namefile, "\n")
  

  rep=rep+1  
}


namefile<-paste("./data/trainD2Cp",NDAG,max(noNodes),type,"RData",sep=".")
trainD2C<-makeModel(allD2C,classifier="RF",EErep=2)
save(file=namefile,list=c("trainD2C","allD2C","descr"))
cat("SAVED ", namefile, "\n")