CatBoost.jl

Julia interface to CatBoost.

Setting up PyCall

Please follow the PyCall guidelines described in PyCall.jl.

We highly recommend using a Julia-specific Python environment to handle dependencies. We recommend that users follow the build instructions in Conda.jl.

If users have installed miniconda on their local machine, we recommend checking out the Julia-specific Python environment (which is usually located at $HOME/.julia/conda/3) and installing catboost there with pip:

pip install catboost

Example

module Regression

using CatBoost

train_data = [[1, 4, 5, 6], [4, 5, 6, 7], [30, 40, 50, 60]]
eval_data = [[2, 4, 6, 8], [1, 4, 50, 60]]
train_labels = [10, 20, 30]

# Initialize CatBoostRegressor
model = CatBoostRegressor(iterations = 2, learning_rate = 1, depth = 2)

# Fit model
fit!(model, train_data, train_labels)

# Get predictions
preds = predict(model, eval_data)

end # module