从 pmdarima.model_selection 导入 train_test_split 不工作

Question

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

以下错误弹出窗口： ModuleNotFoundError：没有名为“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 版本，解决了这个问题。