【发布时间】:2020-07-17 23:05:19
【问题描述】:
我是 RAPIDS 的新手,之前尝试使用 sklearn 复制逻辑回归模型,但现在使用 cuml,但 LogisticRegression() 调用不起作用。
from cuml.linear_model import LogisticRegression, LinearRegression
ols_cuml=LogisticRegression(penalty='l2')
ols_cuml.fit(X,y)
我收到以下错误:
LinearRegression() 加载语句也会出现同样的错误。考虑到 cudf(Pandas for CUDA) 加载良好并且可以在 gpu 数据帧上运行,不知道为什么会发生这种情况。
【问题讨论】:
-
嗨 alf,欢迎来到 SO。你介意分享mcve吗?