From 0926f580b25739495ff1a452b2a12e75ba0dd2f0 Mon Sep 17 00:00:00 2001
From: WANG Zhiwei <48282751+statwangz@users.noreply.github.com>
Date: Sun, 17 Dec 2023 18:47:34 +0800
Subject: [PATCH] Update softImputeCV.R

---
 others/softImputeCV.R | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/others/softImputeCV.R b/others/softImputeCV.R
index cab7bf6..39c3dd3 100644
--- a/others/softImputeCV.R
+++ b/others/softImputeCV.R
@@ -26,7 +26,11 @@ softImputeCrossVal <- function(Y, rank_max = NULL, lambda_range = NULL,
   if (is.null(lambda_range)) {
     hi_fit <- softImpute(Y, rank.max = rank_max)
     lambda_min <- tail(hi_fit$d, 1) / 100
-    lambda_max <- mean((hi_fit$d)[1:2])
+    if(length(hi_fit$d) > 1) {
+      lambda_max <- mean((hi_fit$d)[1:2])
+    } else {
+      lambda_max <- hi_fit$d
+    }
   } else {
     lambda_min <- lambda_range[1]
     lambda_max <- lambda_range[2]