无法从 AutoGluon 导入 TabularPrediction

问题

你尝试从 AutoGluon 导入 TabularPrediction,但收到错误消息。

ImportError: cannot import name 'TabularPrediction' from 'autogluon' (unknown location)

当通过笔记本或作为群集安装的库安装 AutoGluon 时,会发生这种情况。

可通过在笔记本中运行 import 命令来重现此错误:

import autogluon as ag
from autogluon import TabularPrediction as task

原因

AutoGluon v0.0.14 中存在命名空间冲突。

autogluon==0.0.14 安装 "gluoncv > = 0.5.0, < 1.0"。 这会导致安装 gluoncv==0.9.0,从而产生命名空间冲突。

解决方案

AutoGluon v0.0.15 中已解决命名空间冲突。 升级到 AutoGluon v0.0.15 以使用 TabularPrediction

autogluon==0.0.15在从 PyPI 安装 AutoGluon 作为autogluon==0.0.15时指定。

还可通过笔记本安装它。

pip install autogluon==0.0.15 autogluon.tabular "mxnet<2.0.0"

升级到 AutoGluon v0.0.15 后,可成功导入 TabularPrediction

import autogluon as ag
from autogluon import TabularPrediction as task