Преглед на файлове

Fixed problems with numpy arrays.

Kristian Schultz преди 3 години
родител
ревизия
276263c064
променени са 1 файла, в които са добавени 5 реда и са изтрити 1 реда
  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}")