No se puede importar TabularPrediction de AutoGluon

Problema

Está intentando importar TabularPrediction desde AutoGluon, pero recibe un mensaje de error.

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

Esto sucede cuando AutoGluon se instala a través de un cuaderno o como una biblioteca instalada en clúster.

Para reproducir el error, ejecute el comando import en el cuaderno:

import autogluon as ag
from autogluon import TabularPrediction as task

Causa

Hay una colisión de espacios de nombres en AutoGluon v0.0.14.

autogluon==0.0.14 instala 'gluoncv > =0.5.0, < 1.0'. Esto da como resultado gluoncv==0.9.0 la instalación, lo que crea la colisión de espacios de nombres.

Solución

La colisión de espacios de nombres se resolvió en AutoGluon v0.0.15. Actualice a AutoGluon v0.0.15 para usar TabularPrediction .

Especifique autogluon==0.0.15 al instalar AutoGluon como una autogluon==0.0.15 clúster desde PyPI.

También puede instalarlo a través de un cuaderno.

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

Después de actualizar a AutoGluon v0.0.15, puede importar correctamente TabularPrediction .

import autogluon as ag
from autogluon import TabularPrediction as task