Ver código fonte

Fixed problems with numpy arrays.

Kristian Schultz 3 anos atrás
pai
commit
276263c064
1 arquivos alterados com 5 adições e 1 exclusões
  1. 5 1
      fdc/fdc.py

+ 5 - 1
fdc/fdc.py

@@ -128,7 +128,11 @@ class FDC:
 
         for (name, clustering, column_list) in actions:
             if column_list is not None:
-                emb = clustering.fit(data[column_list])
+                if str(type(data)) == "<class 'numpy.ndarray'>":
+                    part = data[:, column_list]
+                else:
+                    part = data[column_list]
+                emb = clustering.fit(part)
                 concat_lists.append(emb)
                 for n in range(emb.shape[1]):
                     concat_column_names.append(f"{name}_UMAP_{n}")