examples/example.imputation.R

In maddin79/darch: Package for Deep Architectures and Restricted Boltzmann Machines

# PIMA2 example using caret imputation, requires packages MASS and RANN
example.imputation <- function(...)
{
  library(mlbench)
  library(RANN)
  data("PimaIndiansDiabetes2")
  
  darch <- darch(diabetes ~ ., PimaIndiansDiabetes2,
    preProc.params = list("method" = c("knnImpute")),
    layers = c(0,100,50,0),
    darch.batchSize = 1,
    darch.returnBestModel.validationErrorFactor = 1,
    darch.fineTuneFunction = "backpropagation",
    darch.unitFunction = c("tanhUnit", "tanhUnit", "softmaxUnit"),
    darch.numEpochs = 10,
    bootstrap = T,
    bootstrap.num = 507,
    ...
  )
  
  darchTest(darch, newdata = PimaIndiansDiabetes2)
  
  darch
}