Update the code

YangLabHKUST · Sep 30, 2023 · edfc2b5 · edfc2b5
1 parent 4ca1c78
commit edfc2b5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/R/mfairInitialization.R b/R/mfairInitialization.R
@@ -34,11 +34,10 @@ createMFAIR <- function(Y, X,
  Y_sparse <- TRUE
  message("The main data matrix Y has been stored in the sparse mode!")
  } else if (Y_sparse == TRUE) { # Y is not in sparse mode, but we want it to be
- obs_tf <- !is.na(Y) # Indicates whether observed or missing
- obs_indices <- which(obs_tf, arr.ind = TRUE) # Indices of observed entries
+ obs_indices <- which(!is.na(Y), arr.ind = TRUE) # Indices of observed entries
  Y <- Matrix::sparseMatrix(
  i = obs_indices[, "row"], j = obs_indices[, "col"],
- x = Y[obs_tf],
+ x = Y[obs_indices],
  dims = c(N, M),
  symmetric = FALSE, triangular = FALSE,
  index1 = TRUE

diff --git a/data-raw/ml10m.R b/data-raw/ml10m.R
@@ -69,8 +69,7 @@ movie_genre_matrix <- t(sapply(
  movies[, 3],
  FUN = function(x) {
  as.integer(all_genres %in%
- unlist(strsplit(x, split = "|", fixed = TRUE))
- )
+ unlist(strsplit(x, split = "|", fixed = TRUE)))
  },
  USE.NAMES = FALSE
 ))