Merge pull request #166 from rggelles/master

Change feature array initialization dtype to uint32
nicodv · Oct 7, 2021 · fefc05f · fefc05f
2 parents a20f6ed + 54d19a4
commit fefc05f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kmodes/util/__init__.py b/kmodes/util/__init__.py
@@ -39,7 +39,7 @@ def encode_features(X, enc_map=None):
     else:
         fit = False
 
-    Xenc = np.zeros(X.shape).astype('int')
+    Xenc = np.zeros(X.shape, dtype='uint32')
     for ii in range(X.shape[1]):
         if fit:
             col_enc = {val: jj for jj, val in enumerate(np.unique(X[:, ii]))