Updates

MichaelLLi · Jan 24, 2020 · f46130f · f46130f · MichaelLLi · Jan 24, 2020
1 parent 2e0a0fa
commit f46130f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "MatrixImpute"
 uuid = "86adb855-6620-4b42-899a-fdd8a58d2ebe"
 authors = ["Michael Li <[email protected]>"]
-version = "0.3.0"
+version = "0.3.1"
 
 [deps]
 Arpack = "7d9fca2a-8960-54d3-9f78-7d1dccf2cb97"

diff --git a/src/MatrixImpute.jl b/src/MatrixImpute.jl
@@ -95,7 +95,7 @@ function fastImputeInner(A,k,γ,lr,B)
     println("Preprocessing Complete")
     A[ismissing.(A)] .= 0
     if M<0.5 && B == nothing
-        output = svds(A,nsv=5)
+        output = svds(A,nsv = k)
         S = diagm(output[1].S)*output[1].Vt
     else
         S = rand(k,p)