如何在自定义预测例程上安装lightgbm

Question

我已经训练了一个LightGBM模型，现在想创建一个自定义预测例程，用于在谷歌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'])

Answer 1

要实现您想要实现的目标，您需要创建一个包含预装的lightGBM的自定义容器，在这里您可以找到关于如何构建和推送自定义容器映像的分步指南。