在 pyenv 中找到正确版本的 python/sklearn 以使用机器学习模型

如何解决在 pyenv 中找到正确版本的 python/sklearn 以使用机器学习模型

我在 Kaggle 上腌制了一个模型，并尝试将其下载到本地运行。我使用诗歌和 pyenv 运行以下命令来创建一个项目：

pyenv local 3.6.6    
poetry new model_api
cd model_test
poetry env use python
poetry add "sklearn>=0.21.3"

但收到以下错误。

如果我只是使用 sklearn 并用诗歌安装它，我在 VS Code 中执行我的代码时会收到此错误。

/bin/python /home/gary/Documents/model_api/model_api/app.py
Traceback (most recent call last):
  File "/home/gary/Documents/model_api/model_api/app.py",line 5,in <module>
    model = pickle.load(f)
ModuleNotFoundError: No module named 'sklearn.ensemble.forest'

这是我试图运行的代码。

import sklearn 
import pickle

f = open('./model/ForestModel','rb')
model = pickle.load(f)

根据我在 Kaggle 上看到的内容，我正在尝试使用 Python 3.6.6 和 sklearn 0.21.3：

如果我尝试使用更新的 Python 版本（如 3.8.10），则会出现相同的错误。我想我错过了一些简单/明显的东西。任何我可以检查的指针或事情将不胜感激。

解决方法

没有您希望安装的版本的软件包 sklearn。我认为您正在寻找 scikit-learn (Docs)。

您可以通过运行以下命令来安装其他依赖项支持的最新版本：

poetry add scikit-learn

或者如果您需要安装特定版本：

poetry add "scikit-learn==0.24.2"

有关其他选项，请查看诗歌文档 here。

,

我会尝试使用 Anaconda 包管理器而不是 pyenv。您可以使用以下代码创建环境：

• conda create -n envName sklearn

他们可能会与 Python 包保持最佳协调，这样您就不会出错。