Skip to content

XpressAI/xai-xgboost

Repository files navigation

Component LibrariesProject Templates
DocsInstallTutorialsDeveloper GuidesContributeBlogDiscord

Build efficient and scalable machine learning models with the Xircuits Component Library for XGBoost integration!


Xircuits Component Library for XGBoost

This library integrates XGBoost into Xircuits workflows, enabling seamless model training, evaluation, and optimization for scalable and efficient machine learning tasks.

Table of Contents

Preview

The Example

XGBoostClassifier

The Result

XGBoostClassifier result

Prerequisites

Before you begin, you will need the following:

  1. Python3.9+.
  2. Xircuits.

Main Xircuits Components

XGBoostBinaryClassifier Component:

Trains a binary classifier with XGBoost, allowing customization of parameters like tree depth, learning rate, and boosting objective.

XGBoostBinaryClassifier

XGBoostMultiClassClassifier Component:

Trains an XGBoost classifier for multi-class problems with configurable tree depth, learning rate, and objective functions. Suitable for tasks requiring efficient handling of multi-class datasets.

XGBoostMultiClassClassifier

XGBoostRegressor Component:

Trains an XGBoost regressor for regression tasks, supporting objectives like squared error and logistic regression. Offers high flexibility and accuracy in modeling numerical targets.

XGBoostRanker Component:

Trains an XGBoost model for ranking tasks using objectives like pairwise and ndcg. Designed for learning-to-rank problems, it optimizes rank-based evaluation metrics.

XGBoostBinaryPredict Component:

Generates predictions from a trained XGBoost binary classifier. Optionally evaluates the accuracy of predictions if the target variable is provided.

Try the Examples

We have provided an example workflow to help you get started with the XGBoost component library. Give it a try and see how you can create custom XGBoost components for your applications.

XGBoostClassifier

This example demonstrates a workflow for training and testing an XGBoost binary classifier using the Iris dataset. It splits the dataset into training and testing sets, trains the classifier, and evaluates its performance with predictions and accuracy.

Installation

To use this component library, ensure that you have an existing Xircuits setup. You can then install the XGBoost library using the component library interface, or through the CLI using:

xircuits install xgboost

You can also do it manually by cloning and installing it:

# base Xircuits directory
git clone https://github.com/XpressAI/xai-xgboost xai_components/xai_xgboost
pip install -r xai_components/xai_xgboost/requirements.txt 

About

Xircuits component library for XGBoost!

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 3

  •  
  •  
  •  

Languages