Packages

exgboost

0.3.1
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
5 Dependencies
Dependants
Files
Activity
exgboost c exgboost include dmatrix.h
Raw

c/exgboost/include/dmatrix.h 

#ifndef EXGBOOST_DMATRIX_H

#define EXGBOOST_DMATRIX_H



#include "utils.h"



ERL_NIF_TERM EXGDMatrixCreateFromFile(ErlNifEnv *env, int argc,

                                      const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixCreateFromMat(ErlNifEnv *env, int argc,

                                     const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixCreateFromSparse(ErlNifEnv *env, int argc,

                                        const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixCreateFromDense(ErlNifEnv *env, int argc,

                                       const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixGetStrFeatureInfo(ErlNifEnv *env, int argc,

                                         const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixSetStrFeatureInfo(ErlNifEnv *env, int argc,

                                         const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixSetDenseInfo(ErlNifEnv *env, int argc,

                                    const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixNumRow(ErlNifEnv *env, int argc,

                              const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixNumCol(ErlNifEnv *env, int argc,

                              const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixNumNonMissing(ErlNifEnv *env, int argc,

                                     const ERL_NIF_TERM argv[]);



ERL_NIF_TERM EXGDMatrixSetInfoFromInterface(ErlNifEnv *env, int argc,

                                            const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGDMatrixSaveBinary(ErlNifEnv *env, int argc,

                                  const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGDMatrixGetFloatInfo(ErlNifEnv *env, int argc,

                                    const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGDMatrixGetUIntInfo(ErlNifEnv *env, int argc,

                                   const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGDMatrixGetDataAsCSR(ErlNifEnv *env, int argc,

                                    const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGDMatrixSliceDMatrix(ErlNifEnv *env, int argc,

                                    const ERL_NIF_TERM argv[]);

ERL_NIF_TERM EXGProxyDMatrixCreate(ErlNifEnv *env, int argc,

                                   const ERL_NIF_TERM argv[]);

#endif