GradientBoosting
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)