Using 'predict' method always returns the same cluster no matter the input

[tjomk]

I am probably doing something wrong.

I have categorical data, in total ~1600 points. I split it to train and test, then clusterise it. It works well. But when I try to fit the train data to any cluster, it always returns 0.

from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=10)

km = kmodes.KModes(n_clusters=50, init='Huang', num_runs=5, verbose=1)
clusters = km.fit_predict(X_train)

The above works fine and the results are great, however the below code returns all zeroes:

test_data_clusters = km.predict(X_test)

Am I misunderstanding the predict function and doing something wrong?

[tjomk]

Sorry, my bad and everything actually works as expected :) Thanks for the awesome library!