tensorflow-gpu 1.15 使用umap报错TypeError: array() takes 1 positional argument but 2 were given解决方案

在tensorflow-gpu 1.15版本中使用umap.umap.fit_transform时,可能会遇到'TypeError: array() takes 1 positional argument but 2 were given'错误。这通常是由于umap版本与tensorflow-gpu版本不兼容导致的。

以下是一些可能的解决方案:

1. 升级umap版本到最新版本

您可以使用以下命令安装最新版本的umap:

pip install --upgrade umap-learn

2. 降低tensorflow-gpu版本

如果升级umap版本无法解决问题,可以尝试将tensorflow-gpu版本降级到1.14或更低版本:

pip install tensorflow-gpu==1.14

3. 检查数据类型

确保输入umap的数据类型正确,可以尝试将数据类型转换为float32:

import numpy as np
import umap.umap_ as umap

X = np.asarray(X).astype('float32')
embedding = umap.UMAP(n_neighbors=10, min_dist=0.1, metric='euclidean').fit_transform(X)

希望以上方法能够帮助您解决问题。如果问题仍然存在,请提供更详细的错误信息和代码片段,以便我们更好地帮助您。

tensorflow-gpu 1.15 使用umap报错TypeError: __array__() takes 1 positional argument but 2 were given解决方案

原文地址: https://www.cveoy.top/t/topic/jr8M 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录