Fit the single factor MFAI model with partially observed main data matrix stored in the sparse mode.
+ Source:R/mfairSingleFactor.R
+ fitSFSparse.Rd
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-05,
+ 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.
+
+