blob: 0adb56c762ae74519a8b0b0f4146b3a62a4e2f4b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
#' Impute missing values
#'
#' Uses the bootstrap EM Algorithm on incomplete data and creates imputed datasets
#'
#' @param data an object of class \code{idframe}
#' @param m the number of imputed datasets to create.
#'
#' @return
#' A list containing the following elements
#'
#' \tabular{ll}{
#' \code{imputations} \tab a list of length \code{m} with an imputed dataset in each
#' entry. Each element is of class \code{idframe} \cr
#' \code{m} \tab an integer indicating the number of imputations run
#' }
#'
#' @details
#' This function uses the \code{\link[Amelia]{amelia}} function, provided by the
#' \pkg{amelia} package, to perform multiple imputation.
#'
#' @references
#' Honaker, J., King, G., Blackwell, M. (2011). Amelia II: A Program for Missing Data.
#' Journal of Statistical Software, 45(7), 1–47. URL http://www.jstatsoft.org/v45/i07/.
#'
#' @seealso \code{\link[Amelia]{amelia}}
#' @export
dataImpute <- function(data,m=5){
# check if the class is correct
if(class(data)!='idframe')
stop("Not an idframe data")
ninputs <- dim(data$input)[2]
dataMatrix <- cbind(data$input,data$output)
a.out <- amelia(dataMatrix,m=5,p2s = 0)
}
|
