Update ml100k.Rmd

YangLabHKUST · Oct 17, 2023 · b081e32 · b081e32
1 parent 6f7311b
commit b081e32
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/vignettes/ml100k.Rmd b/vignettes/ml100k.Rmd
@@ -101,8 +101,7 @@ Next, let's create a permuted movie genre matrix $X^{\text{pmt}}$, where each co
 Then we fit the MFAI model with $Y$ and $X^{\text{pmt}}$ as input.
 
 ```{r}
-n_pmt <- dim(X)[2]
-
+n_pmt <- ncol(X)
 X_pmt <- apply(X,
  MARGIN = 2,
  FUN = function(x) {