Compute Probability of Each True Mediator, for Every Subject
true_classification_prob.Rd
Compute the probability of the latent true mediator \(M \in \{1, 2 \}\) as
\(P(M_i = j | X_i) = \frac{\exp(X_i \beta)}{1 + \exp(X_i \beta)}\)
for each of the \(i = 1, \dots,\) n
subjects.
Arguments
- beta_matrix
A numeric column matrix of estimated regression parameters for the true mediator mechanism,
M
(true mediator) ~X
(predictor matrix of interest), obtained fromCOMMA_EM
,COMMA_PVW
, orCOMMA_OLS
.- x_matrix
A numeric matrix of covariates in the true mediator mechanism.
x_matrix
should not contain an intercept.
Value
true_classification_prob
returns a dataframe containing three columns.
The first column, Subject
, represents the subject ID, from \(1\) to n
,
where n
is the sample size, or equivalently, the number of rows in x_matrix
.
The second column, M
, represents a true, latent mediator category \(M \in \{1, 2 \}\).
The last column, Probability
, is the value of the equation
\(P(M_i = j | X_i) = \frac{\exp(X_i \beta)}{1 + \exp(X_i \beta)}\) computed
for each subject and true, latent mediator category.
Examples
set.seed(123)
sample_size <- 1000
cov1 <- rnorm(sample_size)
cov2 <- rnorm(sample_size, 1, 2)
x_matrix <- matrix(c(cov1, cov2), nrow = sample_size, byrow = FALSE)
estimated_betas <- matrix(c(1, -1, .5), ncol = 1)
P_M <- true_classification_prob(estimated_betas, x_matrix)
head(P_M)
#> Subject M Probability
#> 1 1 1 0.7435833
#> 2 2 1 0.6660164
#> 3 3 1 0.4808373
#> 4 4 1 0.7853830
#> 5 5 1 0.2352985
#> 6 6 1 0.6954044