XGBoostModelEval

XGBoost model evaluation step

SageMaker step type

Processing

Node type

internal (consumes upstream, produces downstream)

Container entry point

xgboost_model_eval.py

Interface file

steps/interfaces/xgboost_model_eval.step.yaml

Compute

Compute kind

xgboost

Functionality

XGBoost model evaluation script. Loads trained model, processes evaluation data, generates performance metrics and visualizations.

Inputs (dependencies)

Input

Type

Required

Compatible producers

model_input

model_artifacts

yes

XGBoostTraining, PyTorchTraining, DummyTraining, XGBoostModel, PyTorchModel

processed_data

processing_output

yes

TabularPreprocessing, CradleDataLoading, RiskTableMapping, CurrencyConversion

Outputs

Output

Type

eval_output

processing_output

metrics_output

processing_output

Consumers (downstream steps)

Steps that declare this step as a compatible input source:

Framework requirements

Package

Version

pandas

>=1.3.0

numpy

>=1.21.0

xgboost

>=1.6.0

matplotlib

>=3.5.0


Back to the Step Catalog