LightGBM

Modo de foco

LightGBM - SageMaker IA da Amazon

Recomendação de EC2 instância da Amazon para o algoritmo LightGBM Cadernos de exemplo

OLightGBM é uma conhecida e eficiente implementação de código aberto do algoritmo baseado em árvores com aumento de gradiente (Gradient Boosting Decision Tree, GBDT). GBDT é um algoritmo de aprendizado supervisionado que tenta prever com precisão uma variável de destino. Para isso, combina um grupo de estimativas de um conjunto de modelos mais simples e mais fracos. O LightGBM usa técnicas adicionais para melhorar significativamente a eficiência e a escalabilidade do GBDT convencional. Esta página inclui informações sobre recomendações de EC2 instâncias da Amazon e exemplos de notebooks para LightGBM.

Recomendação de EC2 instância da Amazon para o algoritmo LightGBM

SageMaker Atualmente, o AI LightGBM oferece suporte ao treinamento de CPU de instância única e de várias instâncias. Para treinamento de CPU em várias instâncias (treinamento distribuído), especifique um valor instance_count maior que 1 ao definir seu Estimador. Para obter mais informações sobre treinamento distribuído com o LightGBM, consulte Treinamento distribuído do Amazon SageMaker AI LightGBM usando o Dask.

LightGBM é um algoritmo de uso intensivo de memória (ao contrário dos de uso intensivo de computação). Portanto, uma instância de computação de uso geral (por exemplo, M5) é uma opção melhor do que uma instância otimizada para computação (por exemplo, C5). Além disso, recomendamos que você tenha memória total suficiente em instâncias específicas para armazenar os dados de treinamento.

Exemplos de cadernos LightGBM

A tabela a seguir descreve uma variedade de exemplos de notebooks que abordam diferentes casos de uso do algoritmo Amazon SageMaker AI LightGBM.

Título do caderno	Descrição
Classificação tabular com Amazon SageMaker AI LightGBM e algoritmo CatBoost	Este notebook demonstra o uso do algoritmo Amazon SageMaker AI LightGBM para treinar e hospedar um modelo de classificação tabular.
Regressão tabular com Amazon SageMaker AI LightGBM e algoritmo CatBoost	Este notebook demonstra o uso do algoritmo Amazon SageMaker AI LightGBM para treinar e hospedar um modelo de regressão tabular.
Treinamento distribuído do Amazon SageMaker AI LightGBM usando o Dask	Este notebook demonstra o treinamento distribuído com o algoritmo Amazon SageMaker AI LightGBM usando a estrutura Dask.

Para obter instruções sobre como criar e acessar instâncias do notebook Jupyter que você pode usar para executar o exemplo em SageMaker IA, consulte. Instâncias do Amazon SageMaker Notebook Depois de criar uma instância do notebook e abri-la, escolha a guia Exemplos de SageMaker IA para ver uma lista de todas as amostras de SageMaker IA. Para abrir um caderno, escolha a guia Uso e depois escolha Criar cópia.