python-3.x - 从 pmdarima.model_selection 导入 train_test_split 不工作

Question

我正在使用 pmdarima 在 python3.7.7 中应用 auto_arima 。使用 from pmdarima.model_selection import train_test_split 拆分数据集时

以下错误弹出窗口：ModuleNotFoundError: No module named 'pmdarima.model_selection'

已经在 Anaconda 中更新了软件包，仍然是同样的错误。

根据 pdmarima 文档，它应该可用： https ://alkaline-ml.com/pmdarima/modules/generated/pmdarima.model_selection.train_test_split.html?highlight=train_test#pmdarima.model_selection.train_test_split

有什么建议吗？

Answer 1

从 Anaconda 中删除 pmdarima 包，然后使用 pip 重新安装。pip install pmdarima 到这个包的 1.6.0 版本，解决了这个问题。