Skip to contents

Fit the single factor MFAI model with partially observed main data matrix stored in the sparse mode.

Usage

fitSFSparse(
  Y,
  X,
  init,
  obs_indices,
  learning_rate,
  tree_parameters,
  stage1 = TRUE,
  iter_max = 5000,
  tol_stage1 = 0.1,
  tol_stage2 = 1e-08,
  verbose_sf = TRUE,
  verbose_loop = TRUE,
  save_tree_list = TRUE
)

Arguments

Y

Main data matrix.

X

A data.frame containing the auxiliary information.

init

A MFAIRSingleFactor object containing the initial parameters for the single factor MAFI model.

obs_indices

Indices of the observed entries in the main data matrix.

learning_rate

Numeric. Parameter for the gradient boosting part.

tree_parameters

A list containing the parameters for the gradient boosting part.

stage1

Logical. Whether to perform fitting algorithm stage1. The greedy algorithm needs while the backfitting algorithm does not need.

iter_max

Integer. Maximum iterations allowed.

tol_stage1

Numeric. Convergence criterion in the first step.

tol_stage2

Numeric. Convergence criterion in the first step.

verbose_sf

Logical. Whether to display the detailed information.

verbose_loop

Logical. Whether to display the detailed information when looping.

save_tree_list

Logical. Whether to save the tree list.

Value

A MFAIRSingleFactor object containing the information about the fitted single factor MFAI model.