如何在自定义预测程序中安装lightgbm？

Question

我已经训练好了一个LightGBM模型，现在想要创建一个自定义的预测例程用于部署到Google AI平台。

在加载模型文件时，它会抛出一个错误，显示未安装lightgbm。 我在setup.py中使用了以下代码，但它没有起作用。

from setuptools import setup, find_packages
DEPENDENCIES = ['lightgbm==3.1.1']

setup(
      name='my_custom_code',
      version='0.4',
      install_requires=DEPENDENCIES,
      include_package_date=True,
      packages=find_packages(),
      description='custom code.',
      scripts=['predictor.py', 'preprocess.py'])

- kanav anand

Answer 1

为了达到您想要的目标，您需要创建一个包含预安装lightGBM的自定义容器。您可以在这里找到一份逐步指南，以了解如何构建和推送自定义容器镜像。请注意保留HTML标签，但不要添加任何解释性文字。

- Eduardo Ortiz