diff --git a/kaggle_requirements.txt b/kaggle_requirements.txt index 6400d8a1..f8e08395 100644 --- a/kaggle_requirements.txt +++ b/kaggle_requirements.txt @@ -69,6 +69,7 @@ keras<3.6 keras-cv keras-nlp keras-tuner +kmodes kornia langid leven diff --git a/tests/test_kmodes.py b/tests/test_kmodes.py new file mode 100644 index 00000000..056a7f71 --- /dev/null +++ b/tests/test_kmodes.py @@ -0,0 +1,20 @@ +import unittest + +import numpy as np +from kmodes.kmodes import KModes + + +class TestKmodes(unittest.TestCase): + def test_plot(self): + data = np.array( + [ + ["A", "B", "C"], + ["B", "C", "A"], + ["C", "A", "B"], + ["A", "C", "B"], + ["A", "A", "B"], + ] + ) + km = KModes(n_clusters=3, init="Huang", n_init=5, verbose=0) + clusters = km.fit_predict(data) + self.assertEqual(len(np.unique(clusters)), 3)