1. Home
  2. /
  3. Lightgbm
  4. /
  5. Simple Classification Example

Simple Classification Example - Lightgbm Typing CST Test

Loading…

Simple Classification Example — Lightgbm Code

A minimal LightGBM example performing classification on a small dataset.

import lightgbm as lgb
import numpy as np
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

# Load dataset
data = load_iris()
X_train, X_test, y_train, y_test = train_test_split(data.data, data.target, test_size=0.2, random_state=42)

# Create dataset for LightGBM
train_data = lgb.Dataset(X_train, label=y_train)

# Define parameters
params = {'objective':'multiclass','num_class':3,'metric':'multi_logloss'}

# Train model
model = lgb.train(params, train_data, num_boost_round=100)

# Predict
y_pred = model.predict(X_test)
y_pred_labels = np.argmax(y_pred, axis=1)
print('Accuracy:', accuracy_score(y_test, y_pred_labels))

Lightgbm Language Guide

LightGBM (Light Gradient Boosting Machine) is a fast, distributed, high-performance gradient boosting framework based on decision tree algorithms, used for ranking, classification, and many other machine learning tasks.

Primary Use Cases

  • ▸Binary and multiclass classification
  • ▸Regression problems
  • ▸Ranking tasks (learning-to-rank)
  • ▸Feature selection and importance analysis
  • ▸Integration in ML pipelines for large-scale structured data

Notable Features

  • ▸Faster training with histogram-based decision tree algorithm
  • ▸Low memory usage compared to XGBoost
  • ▸Supports parallel and GPU learning
  • ▸Handles categorical features directly
  • ▸Scales efficiently with large datasets

Origin & Creator

LightGBM was developed by Microsoft’s DMTK team and released in 2016 to provide a faster and more memory-efficient gradient boosting framework compared to existing solutions.

Industrial Note

LightGBM is widely used in Kaggle competitions, finance, advertising, recommendation systems, and any scenario requiring high-speed gradient boosting on large datasets.

More Lightgbm Typing Exercises

LightGBM Binary Classification ExampleLightGBM Regression ExampleLightGBM with Categorical FeaturesLightGBM Early Stopping ExampleLightGBM Feature Importance ExampleLightGBM Cross Validation ExampleLightGBM Regression with ValidationLightGBM Multi-class Classification Example

Practice Other Languages

CReactPythonC++RustTypeScriptKotlinPHPJavaC#RubyMqlCqlN1qlCypher