Skip to content

jayzuccarelli/GradientBoosting.jl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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)