44 lines
1.1 KiB
C++
44 lines
1.1 KiB
C++
#ifndef XGBOOST_PYTHON_H
|
|
#define XGBOOST_PYTHON_H
|
|
/*!
|
|
* \file xgboost_regrank_data.h
|
|
* \brief python wrapper for xgboost, using ctypes,
|
|
* hides everything behind functions
|
|
* use c style interface
|
|
*/
|
|
#include "../booster/xgboost_data.h"
|
|
/*! \brief type of row entry */
|
|
typedef xgboost::booster::FMatrixS::REntry XGEntry;
|
|
|
|
/*!
|
|
* \brief create a data matrix
|
|
* \return a new data matrix
|
|
*/
|
|
void* XGDMatrixCreate(void);
|
|
/*!
|
|
* \brief free space in data matrix
|
|
*/
|
|
void XGDMatrixFree(void *handle);
|
|
/*!
|
|
* \brief load a data matrix from text file or buffer(if exists)
|
|
* \param handle a instance of data matrix
|
|
* \param fname file name
|
|
*/
|
|
void XGDMatrixLoad(void *handle, const char *fname);
|
|
/*!
|
|
* \brief load a data matrix into binary file
|
|
* \param handle a instance of data matrix
|
|
* \param fname file name
|
|
*/
|
|
void XGDMatrixSaveBinary( void *handle, const char *fname );
|
|
/*!
|
|
* \brief add row
|
|
* \param handle a instance of data matrix
|
|
* \param fname file name
|
|
* \return a new data matrix
|
|
*/
|
|
//void XGDMatrixPush( void *handle, const std::pair<int,> );
|
|
|
|
#endif
|
|
|