Pass item models in a list instead of an MxMatrix
[openmx:openmx.git] / models / nightly / ifa-cai2009.R
1 # This data is from an email:
2 #
3 # Date: Wed, 06 Feb 2013 19:49:24 -0800
4 # From: Li Cai <lcai@ucla.edu>
5 # To: Joshua N Pritikin <jpritikin@pobox.com>
6 # Subject: Re: how did you control item bias in Cai (2010, p. 592) ?
7
8 #options(error = utils::recover)
9 library(OpenMx)
10 library(rpf)
11
12 correct.LL <- 29995.30
13 data.raw <- suppressWarnings(try(read.csv("models/nightly/data/cai2009.csv"), silent=TRUE))
14 if (is(data.raw, "try-error")) data.raw <- read.csv("data/cai2009.csv")
15 data.g1 <- as.data.frame(data.raw[data.raw$G==1, 2:13])
16 data.g2 <- as.data.frame(data.raw[data.raw$G==2, 2:17])
17 if (0) {
18   # for flexmirt
19   write.table(data.g1, "cai2009-g1.csv", quote=FALSE, row.names=FALSE, col.names=FALSE)
20   write.table(data.g2, "cai2009-g2.csv", quote=FALSE, row.names=FALSE, col.names=FALSE)  
21   fm <- read.flexmirt("~/irt/cai2009/cai2009-prm.txt")
22   fm$G1$spec <- NULL
23   fm$G2$spec <- NULL
24 }
25
26 fm <- structure(list(G1 = structure(list(param = structure(c(0.992675,  0.646717, 0, 0, 0.876469, 1.41764, 1.25402, 0, 0, 0.0826927,  1.76547, 1.20309, 0, 0, -0.346706, 2.1951, 0.844399, 0, 0, -0.978301,  1.37774, 0, 1.06694, 0, 0.992373, 1.80365, 0, 0.814109, 0, 0.213559,  2.15718, 0, 1.58086, 0, -0.418129, 1.18201, 0, 1.56533, 0, -1.24173,  1.80474, 0, 0, 1.0774, 0.810718, 2.60754, 0, 0, 1.23507, 0.0598008,  1.01874, 0, 0, 0.724402, -0.294029, 1.68916, 0, 0, 1.37546, -1.13333 ), .Dim = c(5L, 12L), .Dimnames = list(NULL, c("i1", "i2", "i3",  "i4", "i5", "i6", "i7", "i8", "i9", "i10", "i11", "i12"))), mean = structure(c(0.822622,  -0.290462, 0.19672, 0.733993), .Names = c("X6", "X7", "X8", "X9" )), cov = structure(c(0.826046, 0, 0, 0, 0, 1.656, 0, 0, 0, 0,  1.11263, 0, 0, 0, 0, 1.07878), .Dim = c(4L, 4L))), .Names = c("param",  "mean", "cov")),
27                      G2 = structure(list(param = structure(c(0.992675,  0.646717, 0, 0, 0, 0.876469, 1.41764, 1.25402, 0, 0, 0, 0.0826927,  1.76547, 1.20309, 0, 0, 0, -0.346706, 2.1951, 0.844399, 0, 0,  0, -0.978301, 1.37774, 0, 1.06694, 0, 0, 0.992373, 1.80365, 0,  0.814109, 0, 0, 0.213559, 2.15718, 0, 1.58086, 0, 0, -0.418129,  1.18201, 0, 1.56533, 0, 0, -1.24173, 1.80474, 0, 0, 1.0774, 0,  0.810718, 2.60754, 0, 0, 1.23507, 0, 0.0598008, 1.01874, 0, 0,  0.724402, 0, -0.294029, 1.68916, 0, 0, 1.37546, 0, -1.13333,  1.75531, 0, 0, 0, 1.20652, 0.875564, 1.26308, 0, 0, 0, 1.25013,  0.196607, 1.44526, 0, 0, 0, 0.990354, -0.351181, 1.89461, 0,  0, 0, 0.85611, -1.09382), .Dim = c(6L, 16L), .Dimnames = list(     NULL, c("i1", "i2", "i3", "i4", "i5", "i6", "i7", "i8", "i9",      "i10", "i11", "i12", "i13", "i14", "i15", "i16"))), mean = structure(c(0,  0, 0, 0, 0), .Names = c("X6", "X7", "X8", "X9", "X10")), cov = structure(c(1,  0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0,  0, 0, 1), .Dim = c(5L, 5L))), .Names = c("param", "mean", "cov" ))), .Names = c("G1", "G2"))
28
29 for (col in colnames(data.g1)) data.g1[[col]] <- mxFactor(data.g1[[col]], levels=0:1)
30 for (col in colnames(data.g2)) data.g2[[col]] <- mxFactor(data.g2[[col]], levels=0:1)
31
32 mk.model <- function(model.name, data, latent.free) {
33 #   name <- "g1"
34 #   data <- data.g1
35 #   latent.free <- TRUE
36 #   
37   numItems <- dim(data)[2]
38   numPersons <- dim(data)[1]
39   spec <- list()
40   spec[1:numItems] <- rpf.grm(factors = 2)
41   
42   dims <- (1 + numItems/4)
43   design <- matrix(c(rep(1,numItems),
44                      kronecker(2:dims,rep(1,4))), byrow=TRUE, ncol=numItems)
45   
46   ip.mat <- mxMatrix(name="ItemParam", nrow=3, ncol=numItems,
47                      values=c(1.4,1,0),
48                      free=c(TRUE,TRUE,TRUE))
49   
50   for (ix in 1:numItems) {
51     for (px in 1:3) {
52       name <- paste(c('p',ix,',',px), collapse='')
53       ip.mat@labels[px,ix] <- name
54     }
55   }
56   eip.mat <- mxAlgebra(ItemParam, name="EItemParam", fixed=TRUE)
57
58   m.mat <- mxMatrix(name="mean", nrow=1, ncol=dims, values=0, free=latent.free)
59   cov.mat <- mxMatrix(name="cov", nrow=dims, ncol=dims, values=diag(dims),
60                       free=latent.free)
61   
62   m1 <- mxModel(model=model.name, ip.mat, eip.mat, m.mat, cov.mat,
63                 mxMatrix(name="Design", nrow=dim(design)[1], ncol=numItems, values=design),
64                 mxData(observed=data, type="raw"),
65                 mxExpectationBA81(
66                   ItemSpec=spec,
67                   Design="Design",
68                   EItemParam="EItemParam",
69                   mean="mean", cov="cov",
70                   qpoints=21, qwidth=5),
71                 mxFitFunctionBA81(ItemParam="ItemParam", rescale=FALSE))
72   m1
73 }
74
75 groups <- paste("g", 1:2, sep="")
76
77 if (1) {
78   g1 <- mk.model("g1", data.g1, TRUE)
79   g2 <- mk.model("g2", data.g2, FALSE)
80   g1@matrices$ItemParam@values <-
81     rbind(fm$G1$param[1,], apply(fm$G1$param[2:4,], 2, sum), fm$G1$param[5,])
82   g1@matrices$mean@values <- t(fm$G1$mean)
83   g1@matrices$cov@values <- fm$G1$cov
84   g2@matrices$ItemParam@values <-
85     rbind(fm$G2$param[1,], apply(fm$G2$param[2:5,], 2, sum), fm$G2$param[6,])
86   
87   cModel <- mxModel(model="cModel", g1,g2,
88                     mxComputeOnce(paste(groups, 'expectation', sep='.'), context='EM'))
89 #  cModel <- mxOption(cModel, "Number of Threads", 1)
90   for (grp in groups) cModel@submodels[[grp]]@expectation@scores <- 'full'
91   cModel.eap <- mxRun(cModel)
92
93   fm.sco.g1 <- suppressWarnings(try(read.table("models/nightly/data/cai2009-sco-g1.txt"), silent=TRUE))
94   if (is(fm.sco.g1, "try-error")) fm.sco.g1 <- read.table("data/cai2009-sco-g1.txt")
95   fm.sco.g2 <- suppressWarnings(try(read.table("models/nightly/data/cai2009-sco-g2.txt"), silent=TRUE))
96   if (is(fm.sco.g2, "try-error")) fm.sco.g2 <- read.table("data/cai2009-sco-g2.txt")
97   colnames(fm.sco.g1) <- c("grp","id",colnames(cModel.eap@submodels$g1@expectation@scores.out))
98   colnames(fm.sco.g2) <- c("grp","id",colnames(cModel.eap@submodels$g2@expectation@scores.out))
99   
100   scores.g1 <- cModel.eap@submodels$g1@expectation@scores.out
101   omxCheckCloseEnough(as.matrix(fm.sco.g1[,-1:-2]),
102                       scores.g1, 1e-3)
103   omxCheckCloseEnough(as.matrix(fm.sco.g2[,-1:-2]),
104                       cModel.eap@submodels$g2@expectation@scores.out, 1e-3)
105   
106     cModel <- mxModel(cModel,
107                       mxFitFunctionMultigroup(paste(groups, "fitfunction", sep=".")),
108                       mxComputeSequence(steps=list(
109                         mxComputeOnce(paste(groups, 'expectation', sep=".")),
110                         mxComputeOnce('fitfunction'))))
111     cModel <- mxRun(cModel)
112     omxCheckCloseEnough(cModel@output$minimum, correct.LL, .01)
113 }
114
115 if(1) {
116   g1 <- mk.model("g1", data.g1, TRUE)
117   g2 <- mk.model("g2", data.g2, FALSE)
118   grpModel <- mxModel(model="groupModel", g1, g2,
119                       mxFitFunctionMultigroup(paste(groups, "fitfunction", sep=".")),
120                       mxComputeIterate(steps=list(
121                         mxComputeOnce(paste(groups, "EItemParam", sep=".")),
122                         mxComputeOnce(paste(groups, 'expectation', sep='.'), context='EM'),
123                         mxComputeNewtonRaphson(free.set=paste(groups, 'ItemParam', sep=".")),
124                         mxComputeOnce(paste(groups, 'expectation', sep=".")),
125                         mxComputeOnce('fitfunction')
126                       )))
127   
128   #grpModel <- mxOption(grpModel, "Number of Threads", 1)
129   
130   # NPSOL options:
131   grpModel <- mxOption(grpModel, "Analytic Gradients", 'Yes')
132   grpModel <- mxOption(grpModel, "Verify level", '-1')
133   grpModel <- mxOption(grpModel, "Function precision", '1.0E-7')
134   
135   grpModel <- mxRun(grpModel)
136     
137   omxCheckCloseEnough(grpModel@fitfunction@result, correct.LL, .01)
138   omxCheckCloseEnough(grpModel@submodels$g1@matrices$mean@values, t(fm$G1$mean), .01)
139   omxCheckCloseEnough(grpModel@submodels$g1@matrices$cov@values, fm$G1$cov, .01)
140 }