GradientBoosting.jl

The Julia Implementation of Gradient Boosting.
Author jayzuccarelli
Popularity
1 Star
Updated Last
2 Years Ago
Started In
December 2019

GradientBoosting

GitHub version MIT license

The Julia implementation of Gradient Boosting. The package is now an official entry in the Julia Registry and can be installed using Julia's default package manager Pkg.jl.

Installation

Pkg.add("GradientBoosting")

Importing the Library

using GradientBoosting

Using the Package

# Train the model
train_predictions, gb_models = GradientBoosting.fit(y_trn, X_trn, lr, max_depth, number_of_trees)

# Predict on test data
test_predictions = GradientBoosting.predict(y_tst, X_tst, lr, gb_models)

Used By Packages

No packages found.