Packages

exgboost

0.4.0
0.5.1 0.5.0 0.4.0 0.3.2 0.3.1 0.3.0 0.2.1 0.2.0 0.1.0

Elixir bindings for the XGBoost library. `EXGBoost` provides an implementation of XGBoost that works with [Nx](https://hexdocs.pm/nx/Nx.html) tensors.

HexDocs

Current section

Files

Jump to
Readme
9 Versions
6 Dependencies
Dependants
Files
Activity
exgboost c exgboost include booster.h
Raw

c/exgboost/include/booster.h 

#ifndef EXGBOOST_BOOSTER_H

#define EXGBOOST_BOOSTER_H



#include "utils.h"



ERL_NIF_TERM EXGBoosterCreate(ErlNifEnv *env, int argc,

                              const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterBoostedRounds(ErlNifEnv *env, int argc,

                                     const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterSlice(ErlNifEnv *env, int argc,

                             const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterSetParam(ErlNifEnv *env, int argc,

                                const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterGetNumFeature(ErlNifEnv *env, int argc,

                                     const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterUpdateOneIter(ErlNifEnv *env, int argc,

                                     const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterBoostOneIter(ErlNifEnv *env, int argc,

                                    const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterEvalOneIter(ErlNifEnv *env, int argc,

                                   const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterGetAttrNames(ErlNifEnv *env, int argc,

                                    const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterGetAttr(ErlNifEnv *env, int argc,

                               const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterSetAttr(ErlNifEnv *env, int argc,

                               const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterSetStrFeatureInfo(ErlNifEnv *env, int argc,

                                         const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterGetStrFeatureInfo(ErlNifEnv *env, int argc,

                                         const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterFeatureScore(ErlNifEnv *env, int argc,

                                    const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterPredictFromDMatrix(ErlNifEnv *env, int argc,

                                          const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterPredictFromDense(ErlNifEnv *env, int argc,

                                        const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterPredictFromCSR(ErlNifEnv *env, int argc,

                                      const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterLoadModel(ErlNifEnv *env, int argc,

                                 const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterSaveModel(ErlNifEnv *env, int argc,

                                 const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterSerializeToBuffer(ErlNifEnv *env, int argc,

                                         const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterDeserializeFromBuffer(ErlNifEnv *env, int argc,

                                             const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterLoadModelFromBuffer(ErlNifEnv *env, int argc,

                                           const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterSaveModelToBuffer(ErlNifEnv *env, int argc,

                                         const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterSaveJsonConfig(ErlNifEnv *env, int argc,

                                      const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterLoadJsonConfig(ErlNifEnv *env, int argc,

                                      const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGBoosterDumpModelEx(ErlNifEnv *env, int argc,

                                   const ERL_NIF_TERM argv[]);

#endif