Perform the MLE of mixtures of Mallows model with Spearman distance on full and partial rankings via EM algorithms. Partial rankings with arbitrary missing positions are supported.
print method for class "emMSmix".
fitMSmix(
rankings,
n_clust = 1,
n_start = 1,
n_iter = 200,
mc_em = FALSE,
eps = 10^(-6),
init = list(list(rho = NULL, theta = NULL, weights = NULL))[rep(1, n_start)],
plot_log_lik = FALSE,
comp_log_lik_part = FALSE,
plot_log_lik_part = FALSE,
parallel = FALSE,
theta_max = 3,
theta_tol = 1e-05,
theta_tune = 1,
subset = NULL,
item_names = NULL
)# S3 method for emMSmix
print(x, ...)
An object of class "emMSmix", namely a list with the following named components:
modList of named objects describing the best fitted model in terms of maximized log-likelihood over the n_start initializations. See Details.
max_log_likMaximized log-likelihood values for each initialization.
partial_dataLogical: whether the dataset includes some partially-ranked sequences.
convergenceBinary convergence indicators of the EM algorithm for each initialization: 1 = convergence has been achieved, 0 = otherwise.
recordBest log-likelihood values sequentially achieved over the n_start initializations.
em_settingsList of settings used to fit the model.
callThe matched call.
The mod sublist contains the following named objects:
rhoInteger \(G\)\(\times\)\(n\) matrix with the MLEs of the component-specific consensus rankings in each row.
thetaNumeric vector with the MLEs of the \(G\) component-specific precision parameters.
weightsNumeric vector with the MLEs of the \(G\) mixture weights.
z_hatNumeric \(N\)\(\times\)\(G\) matrix of the estimated posterior component membership probabilities. Returned when n_clust > 1, otherwise NULL.
map_classificationInteger vector of \(N\) mixture component memberships based on the MAP allocation from the z_hat matrix. Returned when n_clust > 1, otherwise NULL.
log_likNumeric vector of the log-likelihood values (based on full or augmented rankings) at each iteration.
best_log_likMaximized log-likelihood value (based on full or augmented rankings) of the fitted model.
bicBIC value of the fitted model based on best_log_lik.
log_lik_partNumeric vector of the observed-data log-likelihood values (based on partial rankings) at each iteration. Returned when rankings contains some partial sequences that can be completed with data_augmentation and plot_log_lik_part = TRUE, otherwise NULL. See Details.
best_log_lik_partMaximized observed-data log-likelihood value (based on partial rankings) of the fitted model. Returned when rankings contains some partial sequences that can be completed with data_augmentation, otherwise NULL. See Details.
bic_partBIC value of the fitted model based on best_log_lik_part. Returned when rankings contains some partial sequences that can be completed with data_augmentation, otherwise NULL. See Details.
convBinary convergence indicator of the best fitted model: 1 = convergence has been achieved, 0 = otherwise.
augmented_rankingsInteger \(N\)\(\times\)\(n\) matrix with rankings completed through the Monte Carlo step in each row. Returned when rankings contains some partial sequences and mc_em = TRUE, otherwise NULL.
Integer \(N\)\(\times\)\(n\) matrix or data frame with partial rankings in each row. Missing positions must be coded as NA.
Number of mixture components. Defaults to 1.
Number of starting points. Defaults to 1.
Maximum number of EM iterations. Defaults to 200.
Logical: whether the Monte Carlo EM algorithm must be used for MLE on partial rankings completion, see Details. Ignored when rankings does not contain any partial sequence. Defaults to FALSE.
Positive tolerance value for the convergence of the EM algorithm. Defaults to \(10^{-6}\).
List of n_start lists with the starting values of the parameters to initialize the EM algorithm. Each list must contain three named objects, namely: 1) rho: integer \(G\)\(\times\)\(n\) matrix with the component-specific consensus rankings in each row; 2) theta: numeric vector of \(G\) non-negative component-specific precision parameters; 3) weights: numeric vector of \(G\) positive mixture weights. Defaults to NULL, meaning that the starting points are automatically generated from the uniform distribution.
Logical: whether the iterative log-likelihood values (based on full or augmented rankings) must be plotted. Defaults to FALSE.
Logical: whether the maximized observed-data log-likelihood value (based on partial rankings) must be returned. Ignored when rankings does not contain any partial sequence or data_augmentation cannot be applied. See Details. Defaults to FALSE.
Logical: whether the iterative observed-data log-likelihood values (based on partial rankings) must be plotted. Ignored when rankings does not contain any partial sequence. In the presence of partial rankings, this argument is ignored when comp_log_lik_part = FALSE or data_augmentation cannot be applied. Defaults to FALSE.
Logical: whether parallelization over multiple initializations must be used. Defaults to FALSE.
Positive upper bound for the precision parameters. Defaults to 3.
Positive convergence tolerance for the Mstep on theta. Defaults to \(10^{-5}\).
Positive tuning constant affecting the precision parameters in the Monte Carlo step. Ignored when rankings does not contain any partial sequence or mc_em = FALSE. Defaults to 1.
Optional logical or integer vector specifying the subset of observations, i.e. rows of the rankings, to be kept. Missing values are taken as FALSE.
Character vector for the names of the items. Defaults to NULL, meaning that colnames(rankings) is used and, if not available, item_names is set equal to "Item1","Item2",....
An object of class "emMSmix" returned by fitMSmix.
Further arguments passed to or from other methods (not used).
The EM algorithms are launched from n_start initializations and the best solution in terms of maximized
log-likelihood value (based on full or augmented rankings) is returned.
When mc_em = FALSE, the scheme introduced by Crispino et al. (2023) is performed, where partial
rankings are augmented with all compatible full rankings. This type of data augmentation is
supported up to 10 missing positions in the partial rankings.
When mc_em = TRUE, the - computationally more efficient - Monte Carlo EM algorithm
introduced by Crispino et al. (2024+) is implemented. In the case of a large number
of censored positions and sample sizes, the mc_em = TRUE must be preferred.
Regardless of the fitting method adopted for inference on partial rankings, note that
setting the argument comp_log_lik_part = TRUE for the computation of the
observed-data log-likelihood values (based on partial rankings)
can slow down the procedure in the case of a large number of censored positions and sample sizes.
Crispino M, Mollica C and Modugno L (2024+). MSmix: An R Package for clustering partial rankings via mixtures of Mallows Models with Spearman distance. (submitted)
Crispino M, Mollica C, Astuti V and Tardella L (2023). Efficient and accurate inference for mixtures of Mallows models with Spearman distance. Statistics and Computing, 33(98), DOI: 10.1007/s11222-023-10266-8.
Sørensen Ø, Crispino M, Liu Q and Vitelli V (2020). BayesMallows: An R Package for the Bayesian Mallows Model. The R Journal, 12(1), pages 324--342, DOI: 10.32614/RJ-2020-026.
Beckett LA (1993). Maximum likelihood estimation in Mallows’s model using partially ranked data. In Probability models and statistical analyses for ranking data, pages 92--107. Springer New York.
summary.emMSmix, plot.emMSmix
## Example 1. Fit the 3-component mixture of Mallow models with Spearman distance
## to the Antifragility dataset.
r_antifrag <- ranks_antifragility[, 1:7]
set.seed(123)
mms_fit <- fitMSmix(rankings = r_antifrag, n_clust = 3, n_start = 10)
mms_fit$mod$rho; mms_fit$mod$theta; mms_fit$mod$weights
## Example 2. Fit the Mallow model with Spearman distance
## to simulated partial rankings through data augmentation.
rank_data <- rbind(c(NA, 4, NA, 1, NA), c(NA, NA, NA, NA, 1), c(2, NA, 1, NA, 3),
c(4, 2, 3, 5, 1), c(NA, 4, 1, 3, 2))
mms_fit <- fitMSmix(rankings = rank_data, n_start = 10)
mms_fit$mod$rho; mms_fit$mod$theta
## Example 3. Fit the Mallow model with Spearman distance
## to the Reading genres dataset through Monte Carlo EM.
top5_read <- ranks_read_genres[, 1:11]
mms_fit <- fitMSmix(rankings = top5_read, n_start = 10, mc_em = TRUE)
mms_fit$mod$rho; mms_fit$mod$theta
Run the code above in your browser using DataLab