如何解决JS调用Python脚本时的sklearn模块找不到问题

核心原因 #

终端运行Python脚本时的环境（Python解释器路径、依赖包位置）和Node.js通过spawn创建的子进程环境不一致，导致子进程找不到已安装的sklearn包。

分步解决方案 #

1. 对齐Python解释器路径 #

• 打开终端，运行which python3，得到你终端正常运行脚本的Python路径（比如/usr/local/bin/python3）
• 修改JS代码中的Python路径为上述结果：

const pyprog = spawn('/usr/local/bin/python3', ['sentiment_analysis_script.py', "my new puppy is so cute"]);

2. 给目标Python环境安装sklearn #

用第一步得到的Python绝对路径执行安装命令，确保包安装到子进程会用到的环境里：

/usr/local/bin/python3 -m pip install scikit-learn

3. 确认工作目录匹配 #

Node.js的工作目录可能和终端不同，导致Python脚本找不到.pkl文件或依赖：

• 可选：在Python脚本开头添加代码打印当前工作目录：

import os
print("当前工作目录：", os.getcwd())

• 如果工作目录不对，有两种解决方式：
  • 方式一：在JS中指定工作目录
    在spawn的第三个参数里添加cwd，指向Python脚本和.pkl文件所在的目录：

    const pyprog = spawn('/usr/local/bin/python3', ['sentiment_analysis_script.py', "my new puppy is so cute"], {
      cwd: '/home/your_username/ml_models' // 替换成你的实际目录
    });
    

  • 方式二：在Python脚本中用绝对路径加载文件
    修改加载.pkl的代码为绝对路径：

    with open('/home/your_username/ml_models/note_sentiment_analysis.pkl', 'rb') as f:
        model = pickle.load(f)
    
    with open('/home/your_username/ml_models/note_vectorizer.pkl', 'rb') as f:
        vectorizer = pickle.load(f)
    

4. 验证依赖是否安装成功 #

用目标Python路径执行以下命令，检查scikit-learn是否存在：

/usr/local/bin/python3 -m pip list | grep scikit-learn

如果输出为空，重新执行第二步的安装命令。

内容的提问来源于stack exchange，提问作者dahlda