From 737e59da172976b3f6fd4fd1541a6629ec29e298 Mon Sep 17 00:00:00 2001 From: WANG Zhiwei <48282751+statwangz@users.noreply.github.com> Date: Wed, 20 Sep 2023 02:48:19 +0800 Subject: [PATCH] Store the rating matrix in the sparse mode --- data-raw/ml100k.R | 20 +++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-) diff --git a/data-raw/ml100k.R b/data-raw/ml100k.R index 7605a99..1d9caa4 100644 --- a/data-raw/ml100k.R +++ b/data-raw/ml100k.R @@ -14,10 +14,14 @@ N <- (range(u_data[, 1]))[2] # Movie, range(u_data[, 2]) M <- (range(u_data[, 2]))[2] -u_data_matrix <- matrix(NA, N, M) -for (i in seq_len(dim(u_data)[1])) { - u_data_matrix[u_data[i, 1], u_data[i, 2]] <- u_data[i, 3] -} +u_data_matrix <- Matrix::sparseMatrix( + i = u_data[, 1], j = u_data[, 2], + x = u_data[, 3], + dims = c(N, M), + symmetric = FALSE, triangular = FALSE, + index1 = TRUE, + repr = "C" +) # head(u_data_matrix) ## User information @@ -51,7 +55,13 @@ u_item <- read.csv( # all_genres u_item <- u_item[, c(6:24)] -colnames(u_item) <- c("Unknown", "Action", "Adventure", "Animation", "Children‘s", "Comedy", "Crime", "Documentary", "Drama", "Fantasy", "Film-Noir", "Horror", "Musical", "Mystery", "Romance", "Sci-Fi", "Thriller", "War", "Western") +colnames(u_item) <- c( + "Unknown", "Action", "Adventure", "Animation", + "Children‘s", "Comedy", "Crime", "Documentary", + "Drama", "Fantasy", "Film-Noir", "Horror", + "Musical", "Mystery", "Romance", "Sci-Fi", + "Thriller", "War", "Western" +) # head(u_item) # dim(u_item)