From b081e321b04296ec9bc115b02b35247c1c4ddc43 Mon Sep 17 00:00:00 2001
From: WANG Zhiwei <48282751+statwangz@users.noreply.github.com>
Date: Tue, 17 Oct 2023 21:58:45 +0800
Subject: [PATCH] Update ml100k.Rmd

---
 vignettes/ml100k.Rmd | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/vignettes/ml100k.Rmd b/vignettes/ml100k.Rmd
index cc2404f..1b24fb1 100644
--- 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) {