Skip to content

Commit b143548

Browse files
committed
Rename cluster -> label in KMedoids
1 parent 373f0fb commit b143548

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/transforms/kmedoids.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ function applyfeat(transform::KMedoids, feat, prep)
100100
iter += 1
101101
end
102102

103-
newfeat = (; cluster=labels) |> Tables.materializer(feat)
103+
newfeat = (; label=labels) |> Tables.materializer(feat)
104104

105105
newfeat, nothing
106106
end

test/transforms/kmedoids.jl

+3-3
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88
b = [randn(100); 10 .+ randn(100)]
99
t = Table(; a, b)
1010
n = t |> KMedoids(2; rng)
11-
i1 = findall(isequal(1), n.cluster)
12-
i2 = findall(isequal(2), n.cluster)
11+
i1 = findall(isequal(1), n.label)
12+
i2 = findall(isequal(2), n.label)
1313
@test mean(t.a[i1]) > 5
1414
@test mean(t.b[i1]) > 5
1515
@test mean(t.a[i2]) < 5
@@ -21,5 +21,5 @@
2121
c = ["a", "b", "c"]
2222
t = Table(; a, b, c)
2323
n = t |> KMedoids(3; rng)
24-
@test sort(n.cluster) == [1, 2, 3]
24+
@test sort(n.label) == [1, 2, 3]
2525
end

0 commit comments

Comments
 (0)