Adicionando Labels e chamando convert

Author: yaansz

src/sklearn.jl

@@ -267,6 +267,8 @@ function generate_low_rank_matrix(; n_samples::Int = 100,
                                                        effective_rank = effective_rank,
                                                        tail_strength = tail_strength,
                                                        random_state = random_state)
-   #return convert(features, labels)
+   labels = zeros(size(features, 1))
+    
+   return convert(features, labels)
 
 end

test/runtests.jl

@@ -54,7 +54,8 @@ using Test
                                                     tail_strength = 0.5,
                                                     random_state = 5)
 
+
     @test size(data)[1] == samples
-    @test size(data)[2] == features
+    @test size(data)[2] == features + 1
 
 end