Skip to content

Commit 053da20

Browse files
authored
add PTBLM tests (#165)
1 parent 1023fc5 commit 053da20

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

test/datasets/text.jl

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,17 @@
11

22
@testset "PTBLM" begin
3-
n_obs = 5574
43
n_targets = 1
54
n_features = ()
6-
Tx=String
7-
Ty=String
8-
9-
d = SMSSpamCollection()
10-
11-
test_supervised_array_dataset(d;
12-
n_obs, n_targets, n_features,
13-
Tx, Ty)
5+
Tx=Vector{String}
6+
Ty=Vector{String}
7+
8+
for (n_obs, split) in [(42068, :train), (3761, :test)]
9+
d = PTBLM(split)
10+
11+
test_supervised_array_dataset(d;
12+
n_obs, n_targets, n_features,
13+
Tx, Ty)
14+
end
1415
end
1516

1617

@@ -33,7 +34,7 @@ end
3334
Tx = Vector{Vector{String}}
3435
for (n_obs, split) in [(12543, :train), (2077, :test), (2001, :dev)]
3536
d = UD_English(split)
36-
37+
3738
test_unsupervised_array_dataset(d;
3839
n_obs, n_features, Tx)
3940
end

0 commit comments

Comments
 (0)