LightGBM-MATLAB.rar_foundyt4_lightGBM_matlab lightgbm_termeaa_轻量

#ifndef LIGHTGBM_C_API_H_ #define LIGHTGBM_C_API_H_ //#include <cstdint> //#include <exception> //#include <stdexcept> //#include <cstring> //#include <string> /*! * To avoid type conversion on large data, most of our expose interface support both for float_32 and float_64. * Except following: * 1. gradients and hessians. * 2. Get current score for training data and validation * The reason is because they are called frequently, the type-conversion on them maybe time cost. */ //#include <LightGBM/export.h> #define LIGHTGBM_C_EXPORT //typedef __int32 int32_t; //typedef unsigned __int32 uint32_t; typedef int int32_t; typedef unsigned int uint32_t; typedef __int64 int64_t; typedef unsigned __int64 uint64_t; typedef void* DatasetHandle; typedef void* BoosterHandle; #define C_API_DTYPE_FLOAT32 (0) #define C_API_DTYPE_FLOAT64 (1) #define C_API_DTYPE_INT32 (2) #define C_API_DTYPE_INT64 (3) #define C_API_PREDICT_NORMAL (0) #define C_API_PREDICT_RAW_SCORE (1) #define C_API_PREDICT_LEAF_INDEX (2) /*! * \brief get string message of the last error * all function in this file will return 0 when succeed * and -1 when an error occured, * \return const char* error inforomation */ LIGHTGBM_C_EXPORT const char* LGBM_GetLastError(); // --- start Dataset interface /*! * \brief load data set from file like the command_line LightGBM do * \param filename the name of the file * \param parameters additional parameters * \param reference used to align bin mapper with other dataset, nullptr means don't used * \param out a loaded dataset * \return 0 when succeed, -1 when failure happens */ LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromFile(const char* filename, const char* parameters, const DatasetHandle reference, DatasetHandle* out); /*! * \brief create a empty dataset by sampling data. * \param sample_data sampled data, grouped by the column. * \param sample_indices indices of sampled data. * \param ncol number columns * \param num_per_col Size of each sampling column * \param num_sample_row Number of sampled rows * \param num_total_row number of total rows * \param parameters additional parameters * \param out created dataset * \return 0 when succeed, -1 when failure happens */ LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromSampledColumn(double** sample_data, int** sample_indices, int32_t ncol, const int* num_per_col, int32_t num_sample_row, int32_t num_total_row, const char* parameters, DatasetHandle* out); /*! * \brief create a empty dataset by reference Dataset * \param reference used to align bin mapper * \param num_total_row number of total rows * \param out created dataset * \return 0 when succeed, -1 when failure happens */ LIGHTGBM_C_EXPORT int LGBM_DatasetCreateByReference(const DatasetHandle reference, int64_t num_total_row, DatasetHandle* out); /*! * \brief push data to existing dataset, if nrow + start_row == num_total_row, will call dataset->FinishLoad * \param dataset handle of dataset * \param data pointer to the data space * \param data_type type of data pointer, can be C_API_DTYPE_FLOAT32 or C_API_DTYPE_FLOAT64 * \param nrow number of rows * \param ncol number columns * \param start_row row start index * \return 0 when succeed, -1 when failure happens */ LIGHTGBM_C_EXPORT int LGBM_DatasetPushRows(DatasetHandle dataset, const void* data, int data_type, int32_t nrow, int32_t ncol, int32_t start_row); /*! * \brief push data to existing dataset, if nrow + start_row == num_total_row, will call dataset->FinishLoad * \param dataset handle of dataset * \param indptr pointer to row headers * \param indptr_type type of indptr, can be C_API_DTYPE_INT32 or C_API_DTYPE_INT64 * \param indices findex * \param data fvalue * \param data_type type of data pointer, can be C_API_DTYPE_FLOAT32 or C_API_DTYPE_FLOAT64 * \param nindptr number of rows in the matrix + 1 * \param nelem number of nonzero elements in the matrix * \param num_col number of columns * \param start_row row start index * \return 0 when succeed, -1 when failure happens */ LIGHTGBM_C_EXPORT int LGBM_DatasetPushRowsByCSR(DatasetHandle dataset, const void* indptr, int indptr_type, const int32_t* indices, const void* data, int data_type, int64_t nindptr, int64_t nelem, int64_t num_col, int64_t start_row); /*! * \brief create a dataset from CSR format * \param indptr pointer to row headers * \param indptr_type type of indptr, can be C_API_DTYPE_INT32 or C_API_DTYPE_INT64 * \param indices findex * \param data fvalue * \param data_type type of data pointer, can be C_API_DTYPE_FLOAT32 or C_API_DTYPE_FLOAT64 * \param nindptr number of rows in the matrix + 1 * \param nelem number of nonzero elements in the matrix * \param num_col number of columns * \param parameters additional parameters * \param reference used to align bin mapper with other dataset, nullptr means don't used * \param out created dataset * \return 0 when succeed, -1 when failure happens */ LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromCSR(const void* indptr, int indptr_type, const int32_t* indices, const void* data, int data_type, int64_t nindptr, int64_t nelem, int64_t num_col, const char* parameters, const DatasetHandle reference, DatasetHandle* out); /*! * \brief create a dataset from CSC format * \param col_ptr pointer to col headers * \param col_ptr_type type of col_ptr, can be C_API_DTYPE_INT32 or C_API_DTYPE_INT64 * \param indices findex * \param data fvalue * \param data_type type of data pointer, can be C_API_DTYPE_FLOAT32 or C_API_DTYPE_FLOAT64 * \param ncol_ptr number of cols in the matrix + 1 * \param nelem number of nonzero elements in the matrix * \param num_row number of rows * \param parameters additional parameters * \param reference used to align bin mapper with other dataset, nullptr means don't used * \param out created dataset * \return 0 when succeed, -1 when failure happens */ LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromCSC(const void* col_ptr, int col_ptr_type, const int32_t* indices,