Когда вы указываете переменную Target в задании SageMaker Training?

Question

Я пытаюсь создать алгоритм машинного обучения, следуя этому уроку: Начало работы с Amazon SageMaker

Если я не пропустил что-то в уроке, я не нашел шагов, где мы указываем целевую переменную. Может кто-нибудь объяснить, где / когда мы указываем нашу целевую переменную при создании модели ML с использованием встроенных алгоритмов SageMaker?

Большое спасибо!

Answer 1

Это зависит от научной парадигмы c, которую вы используете в SageMaker:)

• Встроенные алгоритмы SageMaker имеют свои входные спецификации, описанные в соответствующей документации. Например, для SageMaker Linear Learner и SageMaker XGBoost целью считается первый столбец.
• С пользовательским кодом, таким как Bring-Your-Own- Контейнеры Docker или SageMaker Framework (для Sklearn, TF, PyTorch, M XNet), поскольку вы пишете код, вы можете написать любой вид логики c, а целью может быть любой столбец вашего набора данных.

Answer 2

После небольшого исследования я нашел ответ. Если вы используете файл CSV, извлеченный из корзины S3, предполагается, что целевая переменная находится в первом столбце.

Если вам нужна дополнительная информация, вы можете проверить эту часть AWS документации:

Общий формат данных для обучения