Skip to content

Commit 07012c3

Browse files
committed
✅ Add tests for multiclass case
1 parent c38f8b0 commit 07012c3

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

test/test_convert.jl

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,32 @@ using RelationalDatasets
101101
"c(+id).",
102102
]
103103
)
104+
105+
@test RelationalDatasets.from_vector(
106+
[0 1 1; 1 0 2; 2 2 0],
107+
[0, 1, 2],
108+
) == (RelationalDatasets.RelationalDataset((
109+
pos=["v4(id1,0).", "v4(id2,1).", "v4(id3,2)."],
110+
neg=[],
111+
facts=[
112+
"v1(id1,0).",
113+
"v1(id2,1).",
114+
"v1(id3,2).",
115+
"v2(id1,1).",
116+
"v2(id2,0).",
117+
"v2(id3,2).",
118+
"v3(id1,1).",
119+
"v3(id2,2).",
120+
"v3(id3,0).",
121+
])),
122+
[
123+
"v1(+id,#varv1).",
124+
"v2(+id,#varv2).",
125+
"v3(+id,#varv3).",
126+
"v4(+id,#classlabel).",
127+
]
128+
)
129+
104130
end
105131

106132
@testset "Misaligned Classification and Regression" begin

0 commit comments

Comments
 (0)