View a markdown version of this page

Tingkatkan XGBoost Versi 0.90 ke Versi 1.5 - Amazon SageMaker AI

Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.

Tingkatkan XGBoost Versi 0.90 ke Versi 1.5

Jika Anda menggunakan SageMaker Python SDK, untuk meningkatkan pekerjaan XGBoost 0.90 yang ada ke versi 1.5, Anda harus menginstal SDK versi 2.x dan mengubah XGBoost dan parameter menjadi 1.5-1. version framework_version Jika Anda menggunakan Boto3, Anda perlu memperbarui gambar Docker, dan beberapa hyperparameters dan tujuan pembelajaran.

Tingkatkan SageMaker AI Python SDK Versi 1.x ke Versi 2.x

Jika Anda masih menggunakan Versi 1.x dari SageMaker Python SDK, Anda harus meng-upgrade versi 2.x dari SageMaker Python SDK. Untuk informasi tentang SDK SageMaker Python versi terbaru, lihat Menggunakan SDK Python Versi 2.x. SageMaker Untuk menginstal versi terbaru, jalankan:

python -m pip install --upgrade sagemaker

Ubah tag gambar menjadi 1,5-1

Jika Anda menggunakan SageMaker Python SDK dan menggunakan algoritma bawaan XGBoost, ubah parameter versi di. image_uris.retrive

from sagemaker import image_uris image_uris.retrieve(framework="xgboost", region="us-west-2", version="1.5-1") estimator = sagemaker.estimator.Estimator(image_uri=xgboost_container, hyperparameters=hyperparameters, role=sagemaker.get_execution_role(), instance_count=1, instance_type='ml.m5.2xlarge', volume_size=5, # 5 GB output_path=output_path)

Jika Anda menggunakan SageMaker Python SDK dan menggunakan XGBoost sebagai kerangka kerja untuk menjalankan skrip pelatihan khusus Anda, ubah parameter di XGBoost API. framework_version

estimator = XGBoost(entry_point = "your_xgboost_abalone_script.py", framework_version='1.5-1', hyperparameters=hyperparameters, role=sagemaker.get_execution_role(), instance_count=1, instance_type='ml.m5.2xlarge', output_path=output_path)

sagemaker.session.s3_inputdi SageMaker Python SDK versi 1.x telah diubah namanya menjadi. sagemaker.inputs.TrainingInput Anda harus menggunakan sagemaker.inputs.TrainingInput seperti pada contoh berikut.

content_type = "libsvm" train_input = TrainingInput("s3://{}/{}/{}/".format(bucket, prefix, 'train'), content_type=content_type) validation_input = TrainingInput("s3://{}/{}/{}/".format(bucket, prefix, 'validation'), content_type=content_type)

Untuk daftar lengkap perubahan SageMaker Python SDK versi 2.x, lihat Menggunakan Versi 2.x dari Python SDK. SageMaker

Ubah Gambar Docker untuk Boto3

Jika Anda menggunakan Boto3 untuk melatih atau menerapkan model Anda, ubah tag image docker (1, 0.72, 0.90-1 atau 0.90-2) menjadi 1.5-1.

{ "AlgorithmSpecification":: { "TrainingImage": "746614075791.dkr.ecr.us-west-1.amazonaws.com/sagemaker-xgboost:1.5-1" } ... }

Jika Anda menggunakan SageMaker Python SDK untuk mengambil jalur registri, ubah parameternya. version image_uris.retrieve

from sagemaker import image_uris image_uris.retrieve(framework="xgboost", region="us-west-2", version="1.5-1")

Perbarui Hyperparameter dan Tujuan Pembelajaran

Parameter diam telah usang dan tidak lagi tersedia di XGBoost 1.5 dan versi yang lebih baru. Gunakan verbosity sebagai gantinya. Jika Anda menggunakan tujuan reg:linear pembelajaran, itu telah ditinggalkan juga mendukung. reg:squarederror Gunakan reg:squarederror sebagai gantinya.

hyperparameters = { "verbosity": "2", "objective": "reg:squarederror", "num_round": "50", ... } estimator = sagemaker.estimator.Estimator(image_uri=xgboost_container, hyperparameters=hyperparameters, ...)