Go to home
Back
2
1

xgb-classifier

:

XGBoost Classification ModelKit

This ModelKit contains an XGBoost binary classification model trained on sklearn synthetic data, sample test data, and a FastAPI inference script.

Usage

  1. Run python code/inference.py to start the FastAPI server.
  2. Send a POST request to /predict with a JSON payload containing features.
  3. Ensure artifacts/model/xgb_model.joblib and artifacts/data/sample_test_data.csv are in the correct paths.