ImportError：无法从“sklearn.utils.fixes”导入名称“_argmax”

Question

我正在使用线性回归设置机器学习模型，但尝试运行

from sklearn.model_selection import train_test_split 

抛出以下异常：

ImportError: cannot import name '_argmax' from 'sklearn.utils.fixes'

我正在使用 PyCharm (Jupyter notebook) 并尝试重新启动内核。

但是，当我尝试在 Python 文件中从 Pycharm 运行代码时，它可以工作。

Answer 1

我重新安装了 sklearn，然后创建了 conda 虚拟环境。 然后我使用了这个来自 Pycharm 的新创建的 conda venv，代码开始工作，即 PyCharm 不再引发导入错误。