File tree Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Original file line number Diff line number Diff line change @@ -212,6 +212,7 @@ function generate_classification(; n_samples::Int = 100,
212
212
213
213
return convert (features, labels)
214
214
end
215
+
215
216
"""
216
217
function generate_swiss_roll(; n_samples::Int = 100,
217
218
noise::Float64 = 0.0,
@@ -227,11 +228,9 @@ function generate_swiss_roll(; n_samples::Int = 100,
227
228
noise:: Float64 = 0.0 ,
228
229
random_state:: Union{Int,Nothing} = nothing )
229
230
230
-
231
231
(features, labels) = datasets. make_swiss_roll ( n_samples = n_samples,
232
232
noise = noise,
233
233
random_state = random_state)
234
234
235
235
return convert (features, labels)
236
-
237
236
end
Original file line number Diff line number Diff line change @@ -27,7 +27,6 @@ using Test
27
27
@test size (data)[1 ] == samples
28
28
@test size (data)[2 ] == 4
29
29
30
-
31
30
data = SyntheticDatasets. generate_regression (n_samples = samples,
32
31
n_features = features,
33
32
noise = 2.2 ,
@@ -40,16 +39,13 @@ using Test
40
39
n_features = features,
41
40
n_classes = 1 )
42
41
43
-
44
42
@test size (data)[1 ] == samples
45
43
@test size (data)[2 ] == features + 1
46
44
47
45
data = SyntheticDatasets. generate_swiss_roll (n_samples = samples,
48
46
noise = 2.2 ,
49
47
random_state = 5 )
50
48
51
- @test @show size (data)[1 ] == samples
52
- @test size (data)[2 ] == 4
53
-
54
-
49
+ @test @show size (data)[1 ] == samples
50
+ @test size (data)[2 ] == 4
55
51
end
You can’t perform that action at this time.
0 commit comments