Não percebi em em nenhum momento a definição de quais variáveis seriam usadas no modelo e qual seria a variável resposta. Assumi (na minha cabeça) que o default é utilizar a primeira coluna como a resposta e todas as demais como covariáveis. É isto mesmo?
Olá Arthur, tudo bem? Espero que sim!
Desculpe pela demora em retornar.
É exatamente isso que você disse Arthur. A primeira variável é tida como a variável alvo, enquanto as outras variáveis serão entendidas como variáveis explicativas.
Você pode checar essa informação na seção Note na documentação do XGBoost do SageMaker.
Esse padrão serve para as funções do SageMaker, mas caso esteja utilizando a bibioteca do scikit learn por exemplo, poderá configurar manualmente.
Bons estudos!