浏览代码

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}")