A library to build Gradient boosted trees for ordinal labels
Project description
OrdinalGBT
Introduction
OrdinalGBT, which stands for Ordinal gradient boosted trees, is a Python package that implements an ordinal regression loss function using the lightGBM framework. Ordinal regression is a type of regression analysis used for predicting an ordinal variable, i.e. a variable that can be sorted in some order. LightGBM is a gradient boosting framework that uses tree-based learning algorithms and is designed to be distributed and efficient.
Installation
You can install OrdinalGBT using pip:
pip install ordinalgbt
Usage
Here are a few examples on how to use the LGBMOrdinal
class:
- Fitting the model
from ordinalgbt.lgb import LGBMOrdinal
import numpy as np
# Create the model
model = LGBMOrdinal()
# Generate some data
X = np.random.rand(100, 10)
y = np.random.randint(0, 3, 100)
# Fit the model
model.fit(X, y)
- Predicting with the model
After fitting the model, you can use it to make predictions:
# Generate some new data
X_new = np.random.rand(10, 10)
# Use the model to make predictions
# the .predict method returns the class prediction rather than raw score or
# probabilities
y_pred = model.predict(X_new)
print(y_pred)
- Predicting probabilities with the model
The predict_proba
method can be used to get the probabilities of each class:
# Use the model to predict probabilities
y_proba = model.predict_proba(X_new)
print(y_proba)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for ordinalgbt-0.1.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | cd8c7e45c53af89c28227d65d0837bbc27179b9d8971c0df972d999133bd2e08 |
|
MD5 | db1095ba81b44284d029774413658d7a |
|
BLAKE2b-256 | d82a6bdd43a3b81faabd1c4074881b258e436ec91fd6585c11aaa952f22350d7 |