独自コンテナを作成してSageMaker上で機械学習モデルの訓練・デプロイ・推論を行う
SageMakerではPyTorchやTensorflowといった機械学習でよく用いられるフレームワークをビルトインで提供しており、これらを用いて簡単に機械学習を実装することが出来るようようになっています。一方で、ビルトインで提供されていないアルゴリズムを利用したい場合のために、独自のコンテナを作成・利用する方法も提供しています。
今回はKaggleのTitanic生存予測を、独自コンテナを作成して行ってみます。本記事で扱うアルゴリズムはXGBoostであり、これはビルトイ