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
- Run
python code/inference.py
to start the FastAPI server. - Send a POST request to
/predict
with a JSON payload containing features. - Ensure
artifacts/model/xgb_model.joblib
andartifacts/data/sample_test_data.csv
are in the correct paths.