Skip to content

GitLab

Commit 46818ac6 authored by cbecker's avatar cbecker Committed by Guolin Ke
Browse files

Export low level interface API for MSVC (#203)

parent e5b94fbe
Hide whitespace changes
Inline Side-by-side
Showing with 133 additions and 121 deletions
include/LightGBM/boosting.h
View file @ 46818ac6
...@@ -17,7 +17,7 @@ class Metric; ...@@ -17,7 +17,7 @@ class Metric;
/*! /*!
* \brief The interface for Boosting * \brief The interface for Boosting
*/ */
class Boosting { class LIGHTGBM_EXPORT Boosting {
public: public:
/*! \brief virtual destructor */ /*! \brief virtual destructor */
virtual ~Boosting() {} virtual ~Boosting() {}
......
include/LightGBM/c_api.h
View file @ 46818ac6
...@@ -14,17 +14,7 @@ ...@@ -14,17 +14,7 @@
* The reason is because they are called frequently, the type-conversion on them maybe time cost. * The reason is because they are called frequently, the type-conversion on them maybe time cost.
*/ */
#ifdef __cplusplus #include <LightGBM/export.h>
#define DLL_EXTERN_C extern "C"
#else
#define DLL_EXTERN_C
#endif
#ifdef _MSC_VER
#define DllExport DLL_EXTERN_C __declspec(dllexport)
#else
#define DllExport DLL_EXTERN_C
#endif
typedef void* DatasetHandle; typedef void* DatasetHandle;
typedef void* BoosterHandle; typedef void* BoosterHandle;
...@@ -44,7 +34,7 @@ typedef void* BoosterHandle; ...@@ -44,7 +34,7 @@ typedef void* BoosterHandle;
* and -1 when an error occured, * and -1 when an error occured,
* \return const char* error inforomation * \return const char* error inforomation
*/ */
DllExport const char* LGBM_GetLastError(); LIGHTGBM_C_EXPORT const char* LGBM_GetLastError();
// --- start Dataset interface // --- start Dataset interface
...@@ -57,7 +47,7 @@ DllExport const char* LGBM_GetLastError(); ...@@ -57,7 +47,7 @@ DllExport const char* LGBM_GetLastError();
* \param out a loaded dataset * \param out a loaded dataset
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetCreateFromFile(const char* filename, LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromFile(const char* filename,
const char* parameters, const char* parameters,
const DatasetHandle reference, const DatasetHandle reference,
DatasetHandle* out); DatasetHandle* out);
...@@ -77,7 +67,7 @@ DllExport int LGBM_DatasetCreateFromFile(const char* filename, ...@@ -77,7 +67,7 @@ DllExport int LGBM_DatasetCreateFromFile(const char* filename,
* \param out created dataset * \param out created dataset
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetCreateFromCSR(const void* indptr, LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromCSR(const void* indptr,
int indptr_type, int indptr_type,
const int32_t* indices, const int32_t* indices,
const void* data, const void* data,
...@@ -104,7 +94,7 @@ DllExport int LGBM_DatasetCreateFromCSR(const void* indptr, ...@@ -104,7 +94,7 @@ DllExport int LGBM_DatasetCreateFromCSR(const void* indptr,
* \param out created dataset * \param out created dataset
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetCreateFromCSC(const void* col_ptr, LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromCSC(const void* col_ptr,
int col_ptr_type, int col_ptr_type,
const int32_t* indices, const int32_t* indices,
const void* data, const void* data,
...@@ -128,7 +118,7 @@ DllExport int LGBM_DatasetCreateFromCSC(const void* col_ptr, ...@@ -128,7 +118,7 @@ DllExport int LGBM_DatasetCreateFromCSC(const void* col_ptr,
* \param out created dataset * \param out created dataset
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetCreateFromMat(const void* data, LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromMat(const void* data,
int data_type, int data_type,
int32_t nrow, int32_t nrow,
int32_t ncol, int32_t ncol,
...@@ -146,7 +136,7 @@ DllExport int LGBM_DatasetCreateFromMat(const void* data, ...@@ -146,7 +136,7 @@ DllExport int LGBM_DatasetCreateFromMat(const void* data,
* \param out subset of data * \param out subset of data
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetGetSubset( LIGHTGBM_C_EXPORT int LGBM_DatasetGetSubset(
const DatasetHandle handle, const DatasetHandle handle,
const int32_t* used_row_indices, const int32_t* used_row_indices,
int32_t num_used_row_indices, int32_t num_used_row_indices,
...@@ -160,7 +150,7 @@ DllExport int LGBM_DatasetGetSubset( ...@@ -160,7 +150,7 @@ DllExport int LGBM_DatasetGetSubset(
* \param num_feature_names number of feature names * \param num_feature_names number of feature names
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetSetFeatureNames( LIGHTGBM_C_EXPORT int LGBM_DatasetSetFeatureNames(
DatasetHandle handle, DatasetHandle handle,
const char** feature_names, const char** feature_names,
int num_feature_names); int num_feature_names);
...@@ -173,7 +163,7 @@ DllExport int LGBM_DatasetSetFeatureNames( ...@@ -173,7 +163,7 @@ DllExport int LGBM_DatasetSetFeatureNames(
* \param num_feature_names number of feature names * \param num_feature_names number of feature names
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetGetFeatureNames( LIGHTGBM_C_EXPORT int LGBM_DatasetGetFeatureNames(
DatasetHandle handle, DatasetHandle handle,
char** feature_names, char** feature_names,
int* num_feature_names); int* num_feature_names);
...@@ -183,7 +173,7 @@ DllExport int LGBM_DatasetGetFeatureNames( ...@@ -183,7 +173,7 @@ DllExport int LGBM_DatasetGetFeatureNames(
* \brief free space for dataset * \brief free space for dataset
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetFree(DatasetHandle handle); LIGHTGBM_C_EXPORT int LGBM_DatasetFree(DatasetHandle handle);
/*! /*!
* \brief save dateset to binary file * \brief save dateset to binary file
...@@ -191,7 +181,7 @@ DllExport int LGBM_DatasetFree(DatasetHandle handle); ...@@ -191,7 +181,7 @@ DllExport int LGBM_DatasetFree(DatasetHandle handle);
* \param filename file name * \param filename file name
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetSaveBinary(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetSaveBinary(DatasetHandle handle,
const char* filename); const char* filename);
/*! /*!
...@@ -205,7 +195,7 @@ DllExport int LGBM_DatasetSaveBinary(DatasetHandle handle, ...@@ -205,7 +195,7 @@ DllExport int LGBM_DatasetSaveBinary(DatasetHandle handle,
* \param type C_API_DTYPE_FLOAT32 or C_API_DTYPE_INT32 * \param type C_API_DTYPE_FLOAT32 or C_API_DTYPE_INT32
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetSetField(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetSetField(DatasetHandle handle,
const char* field_name, const char* field_name,
const void* field_data, const void* field_data,
int num_element, int num_element,
...@@ -220,7 +210,7 @@ DllExport int LGBM_DatasetSetField(DatasetHandle handle, ...@@ -220,7 +210,7 @@ DllExport int LGBM_DatasetSetField(DatasetHandle handle,
* \param out_type C_API_DTYPE_FLOAT32 or C_API_DTYPE_INT32 * \param out_type C_API_DTYPE_FLOAT32 or C_API_DTYPE_INT32
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetGetField(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetGetField(DatasetHandle handle,
const char* field_name, const char* field_name,
int* out_len, int* out_len,
const void** out_ptr, const void** out_ptr,
...@@ -232,7 +222,7 @@ DllExport int LGBM_DatasetGetField(DatasetHandle handle, ...@@ -232,7 +222,7 @@ DllExport int LGBM_DatasetGetField(DatasetHandle handle,
* \param out The address to hold number of data * \param out The address to hold number of data
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetGetNumData(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetGetNumData(DatasetHandle handle,
int* out); int* out);
/*! /*!
...@@ -241,7 +231,7 @@ DllExport int LGBM_DatasetGetNumData(DatasetHandle handle, ...@@ -241,7 +231,7 @@ DllExport int LGBM_DatasetGetNumData(DatasetHandle handle,
* \param out The output of number of features * \param out The output of number of features
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_DatasetGetNumFeature(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetGetNumFeature(DatasetHandle handle,
int* out); int* out);
// --- start Booster interfaces // --- start Booster interfaces
...@@ -253,7 +243,7 @@ DllExport int LGBM_DatasetGetNumFeature(DatasetHandle handle, ...@@ -253,7 +243,7 @@ DllExport int LGBM_DatasetGetNumFeature(DatasetHandle handle,
* \prama out handle of created Booster * \prama out handle of created Booster
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterCreate(const DatasetHandle train_data, LIGHTGBM_C_EXPORT int LGBM_BoosterCreate(const DatasetHandle train_data,
const char* parameters, const char* parameters,
BoosterHandle* out); BoosterHandle* out);
...@@ -264,7 +254,7 @@ DllExport int LGBM_BoosterCreate(const DatasetHandle train_data, ...@@ -264,7 +254,7 @@ DllExport int LGBM_BoosterCreate(const DatasetHandle train_data,
* \param out handle of created Booster * \param out handle of created Booster
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterCreateFromModelfile( LIGHTGBM_C_EXPORT int LGBM_BoosterCreateFromModelfile(
const char* filename, const char* filename,
int* out_num_iterations, int* out_num_iterations,
BoosterHandle* out); BoosterHandle* out);
...@@ -275,7 +265,7 @@ DllExport int LGBM_BoosterCreateFromModelfile( ...@@ -275,7 +265,7 @@ DllExport int LGBM_BoosterCreateFromModelfile(
* \param handle handle to be freed * \param handle handle to be freed
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterFree(BoosterHandle handle); LIGHTGBM_C_EXPORT int LGBM_BoosterFree(BoosterHandle handle);
/*! /*!
* \brief Merge model in two booster to first handle * \brief Merge model in two booster to first handle
...@@ -283,7 +273,7 @@ DllExport int LGBM_BoosterFree(BoosterHandle handle); ...@@ -283,7 +273,7 @@ DllExport int LGBM_BoosterFree(BoosterHandle handle);
* \param other_handle * \param other_handle
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterMerge(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterMerge(BoosterHandle handle,
BoosterHandle other_handle); BoosterHandle other_handle);
/*! /*!
...@@ -292,7 +282,7 @@ DllExport int LGBM_BoosterMerge(BoosterHandle handle, ...@@ -292,7 +282,7 @@ DllExport int LGBM_BoosterMerge(BoosterHandle handle,
* \param valid_data validation data set * \param valid_data validation data set
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterAddValidData(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterAddValidData(BoosterHandle handle,
const DatasetHandle valid_data); const DatasetHandle valid_data);
/*! /*!
...@@ -301,7 +291,7 @@ DllExport int LGBM_BoosterAddValidData(BoosterHandle handle, ...@@ -301,7 +291,7 @@ DllExport int LGBM_BoosterAddValidData(BoosterHandle handle,
* \param train_data training data set * \param train_data training data set
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterResetTrainingData(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterResetTrainingData(BoosterHandle handle,
const DatasetHandle train_data); const DatasetHandle train_data);
/*! /*!
...@@ -310,7 +300,7 @@ DllExport int LGBM_BoosterResetTrainingData(BoosterHandle handle, ...@@ -310,7 +300,7 @@ DllExport int LGBM_BoosterResetTrainingData(BoosterHandle handle,
* \param parameters format: 'key1=value1 key2=value2' * \param parameters format: 'key1=value1 key2=value2'
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterResetParameter(BoosterHandle handle, const char* parameters); LIGHTGBM_C_EXPORT int LGBM_BoosterResetParameter(BoosterHandle handle, const char* parameters);
/*! /*!
* \brief Get number of class * \brief Get number of class
...@@ -318,7 +308,7 @@ DllExport int LGBM_BoosterResetParameter(BoosterHandle handle, const char* param ...@@ -318,7 +308,7 @@ DllExport int LGBM_BoosterResetParameter(BoosterHandle handle, const char* param
* \param out_len number of class * \param out_len number of class
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterGetNumClasses(BoosterHandle handle, int* out_len); LIGHTGBM_C_EXPORT int LGBM_BoosterGetNumClasses(BoosterHandle handle, int* out_len);
/*! /*!
* \brief update the model in one round * \brief update the model in one round
...@@ -326,7 +316,7 @@ DllExport int LGBM_BoosterGetNumClasses(BoosterHandle handle, int* out_len); ...@@ -326,7 +316,7 @@ DllExport int LGBM_BoosterGetNumClasses(BoosterHandle handle, int* out_len);
* \param is_finished 1 means finised(cannot split any more) * \param is_finished 1 means finised(cannot split any more)
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterUpdateOneIter(BoosterHandle handle, int* is_finished); LIGHTGBM_C_EXPORT int LGBM_BoosterUpdateOneIter(BoosterHandle handle, int* is_finished);
/*! /*!
* \brief update the model, by directly specify gradient and second order gradient, * \brief update the model, by directly specify gradient and second order gradient,
...@@ -337,7 +327,7 @@ DllExport int LGBM_BoosterUpdateOneIter(BoosterHandle handle, int* is_finished); ...@@ -337,7 +327,7 @@ DllExport int LGBM_BoosterUpdateOneIter(BoosterHandle handle, int* is_finished);
* \param is_finished 1 means finised(cannot split any more) * \param is_finished 1 means finised(cannot split any more)
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterUpdateOneIterCustom(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterUpdateOneIterCustom(BoosterHandle handle,
const float* grad, const float* grad,
const float* hess, const float* hess,
int* is_finished); int* is_finished);
...@@ -347,21 +337,21 @@ DllExport int LGBM_BoosterUpdateOneIterCustom(BoosterHandle handle, ...@@ -347,21 +337,21 @@ DllExport int LGBM_BoosterUpdateOneIterCustom(BoosterHandle handle,
* \param handle handle * \param handle handle
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterRollbackOneIter(BoosterHandle handle); LIGHTGBM_C_EXPORT int LGBM_BoosterRollbackOneIter(BoosterHandle handle);
/*! /*!
* \brief Get iteration of current boosting rounds * \brief Get iteration of current boosting rounds
* \param out_iteration iteration of boosting rounds * \param out_iteration iteration of boosting rounds
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterGetCurrentIteration(BoosterHandle handle, int* out_iteration); LIGHTGBM_C_EXPORT int LGBM_BoosterGetCurrentIteration(BoosterHandle handle, int* out_iteration);
/*! /*!
* \brief Get number of eval * \brief Get number of eval
* \param out_len total number of eval results * \param out_len total number of eval results
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterGetEvalCounts(BoosterHandle handle, int* out_len); LIGHTGBM_C_EXPORT int LGBM_BoosterGetEvalCounts(BoosterHandle handle, int* out_len);
/*! /*!
* \brief Get Name of eval * \brief Get Name of eval
...@@ -369,7 +359,7 @@ DllExport int LGBM_BoosterGetEvalCounts(BoosterHandle handle, int* out_len); ...@@ -369,7 +359,7 @@ DllExport int LGBM_BoosterGetEvalCounts(BoosterHandle handle, int* out_len);
* \param out_strs names of eval result, need to pre-allocate memory before call this * \param out_strs names of eval result, need to pre-allocate memory before call this
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterGetEvalNames(BoosterHandle handle, int* out_len, char** out_strs); LIGHTGBM_C_EXPORT int LGBM_BoosterGetEvalNames(BoosterHandle handle, int* out_len, char** out_strs);
/*! /*!
* \brief get evaluation for training data and validation data * \brief get evaluation for training data and validation data
...@@ -381,7 +371,7 @@ Note: 1. you should call LGBM_BoosterGetEvalNames first to get the name of evalu ...@@ -381,7 +371,7 @@ Note: 1. you should call LGBM_BoosterGetEvalNames first to get the name of evalu
* \param out_result float arrary contains result * \param out_result float arrary contains result
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterGetEval(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterGetEval(BoosterHandle handle,
int data_idx, int data_idx,
int* out_len, int* out_len,
double* out_results); double* out_results);
...@@ -395,7 +385,7 @@ Note: should pre-allocate memory for out_result, its length is equal to num_cla ...@@ -395,7 +385,7 @@ Note: should pre-allocate memory for out_result, its length is equal to num_cla
* \param out_len len of output result * \param out_len len of output result
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterGetNumPredict(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterGetNumPredict(BoosterHandle handle,
int data_idx, int data_idx,
int64_t* out_len); int64_t* out_len);
...@@ -409,7 +399,7 @@ Note: should pre-allocate memory for out_result, its length is equal to num_cla ...@@ -409,7 +399,7 @@ Note: should pre-allocate memory for out_result, its length is equal to num_cla
* \param out_result used to set a pointer to array, should allocate memory before call this function * \param out_result used to set a pointer to array, should allocate memory before call this function
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterGetPredict(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterGetPredict(BoosterHandle handle,
int data_idx, int data_idx,
int64_t* out_len, int64_t* out_len,
double* out_result); double* out_result);
...@@ -427,7 +417,7 @@ DllExport int LGBM_BoosterGetPredict(BoosterHandle handle, ...@@ -427,7 +417,7 @@ DllExport int LGBM_BoosterGetPredict(BoosterHandle handle,
* \param result_filename filename of result file * \param result_filename filename of result file
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterPredictForFile(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterPredictForFile(BoosterHandle handle,
const char* data_filename, const char* data_filename,
int data_has_header, int data_has_header,
int predict_type, int predict_type,
...@@ -446,7 +436,7 @@ DllExport int LGBM_BoosterPredictForFile(BoosterHandle handle, ...@@ -446,7 +436,7 @@ DllExport int LGBM_BoosterPredictForFile(BoosterHandle handle,
* \param out_len lenght of prediction * \param out_len lenght of prediction
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterCalcNumPredict(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterCalcNumPredict(BoosterHandle handle,
int num_row, int num_row,
int predict_type, int predict_type,
int num_iteration, int num_iteration,
...@@ -475,7 +465,7 @@ DllExport int LGBM_BoosterCalcNumPredict(BoosterHandle handle, ...@@ -475,7 +465,7 @@ DllExport int LGBM_BoosterCalcNumPredict(BoosterHandle handle,
* \param out_result used to set a pointer to array, should allocate memory before call this function * \param out_result used to set a pointer to array, should allocate memory before call this function
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterPredictForCSR(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterPredictForCSR(BoosterHandle handle,
const void* indptr, const void* indptr,
int indptr_type, int indptr_type,
const int32_t* indices, const int32_t* indices,
...@@ -512,7 +502,7 @@ DllExport int LGBM_BoosterPredictForCSR(BoosterHandle handle, ...@@ -512,7 +502,7 @@ DllExport int LGBM_BoosterPredictForCSR(BoosterHandle handle,
* \param out_result used to set a pointer to array, should allocate memory before call this function * \param out_result used to set a pointer to array, should allocate memory before call this function
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterPredictForCSC(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterPredictForCSC(BoosterHandle handle,
const void* col_ptr, const void* col_ptr,
int col_ptr_type, int col_ptr_type,
const int32_t* indices, const int32_t* indices,
...@@ -546,7 +536,7 @@ DllExport int LGBM_BoosterPredictForCSC(BoosterHandle handle, ...@@ -546,7 +536,7 @@ DllExport int LGBM_BoosterPredictForCSC(BoosterHandle handle,
* \param out_result used to set a pointer to array, should allocate memory before call this function * \param out_result used to set a pointer to array, should allocate memory before call this function
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterPredictForMat(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterPredictForMat(BoosterHandle handle,
const void* data, const void* data,
int data_type, int data_type,
int32_t nrow, int32_t nrow,
...@@ -564,7 +554,7 @@ DllExport int LGBM_BoosterPredictForMat(BoosterHandle handle, ...@@ -564,7 +554,7 @@ DllExport int LGBM_BoosterPredictForMat(BoosterHandle handle,
* \param filename file name * \param filename file name
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterSaveModel(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterSaveModel(BoosterHandle handle,
int num_iteration, int num_iteration,
const char* filename); const char* filename);
...@@ -577,7 +567,7 @@ DllExport int LGBM_BoosterSaveModel(BoosterHandle handle, ...@@ -577,7 +567,7 @@ DllExport int LGBM_BoosterSaveModel(BoosterHandle handle,
* \param out_str json format string of model, need to pre-allocate memory before call this * \param out_str json format string of model, need to pre-allocate memory before call this
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterDumpModel(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterDumpModel(BoosterHandle handle,
int num_iteration, int num_iteration,
int buffer_len, int buffer_len,
int* out_len, int* out_len,
...@@ -591,7 +581,7 @@ DllExport int LGBM_BoosterDumpModel(BoosterHandle handle, ...@@ -591,7 +581,7 @@ DllExport int LGBM_BoosterDumpModel(BoosterHandle handle,
* \param out_val out result * \param out_val out result
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterGetLeafValue(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterGetLeafValue(BoosterHandle handle,
int tree_idx, int tree_idx,
int leaf_idx, int leaf_idx,
double* out_val); double* out_val);
...@@ -604,7 +594,7 @@ DllExport int LGBM_BoosterGetLeafValue(BoosterHandle handle, ...@@ -604,7 +594,7 @@ DllExport int LGBM_BoosterGetLeafValue(BoosterHandle handle,
* \param val leaf value * \param val leaf value
* \return 0 when succeed, -1 when failure happens * \return 0 when succeed, -1 when failure happens
*/ */
DllExport int LGBM_BoosterSetLeafValue(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterSetLeafValue(BoosterHandle handle,
int tree_idx, int tree_idx,
int leaf_idx, int leaf_idx,
double val); double val);
......
include/LightGBM/config.h
View file @ 46818ac6
...@@ -5,6 +5,7 @@ ...@@ -5,6 +5,7 @@
#include <LightGBM/utils/log.h> #include <LightGBM/utils/log.h>
#include <LightGBM/meta.h> #include <LightGBM/meta.h>
#include <LightGBM/export.h>
#include <vector> #include <vector>
#include <string> #include <string>
...@@ -123,7 +124,7 @@ public: ...@@ -123,7 +124,7 @@ public:
* And add an prefix "name:" while using column name * And add an prefix "name:" while using column name
* Note: when using Index, it dosen't count the label index */ * Note: when using Index, it dosen't count the label index */
std::string categorical_column = ""; std::string categorical_column = "";
void Set(const std::unordered_map<std::string, std::string>& params) override; LIGHTGBM_EXPORT void Set(const std::unordered_map<std::string, std::string>& params) override;
}; };
/*! \brief Config for objective function */ /*! \brief Config for objective function */
...@@ -145,7 +146,7 @@ public: ...@@ -145,7 +146,7 @@ public:
int num_class = 1; int num_class = 1;
// Balancing of positive and negative weights // Balancing of positive and negative weights
double scale_pos_weight = 1.0f; double scale_pos_weight = 1.0f;
void Set(const std::unordered_map<std::string, std::string>& params) override; LIGHTGBM_EXPORT void Set(const std::unordered_map<std::string, std::string>& params) override;
}; };
/*! \brief Config for metrics interface*/ /*! \brief Config for metrics interface*/
...@@ -158,7 +159,7 @@ public: ...@@ -158,7 +159,7 @@ public:
double fair_c = 1.0f; double fair_c = 1.0f;
std::vector<double> label_gain; std::vector<double> label_gain;
std::vector<int> eval_at; std::vector<int> eval_at;
void Set(const std::unordered_map<std::string, std::string>& params) override; LIGHTGBM_EXPORT void Set(const std::unordered_map<std::string, std::string>& params) override;
}; };
...@@ -182,7 +183,7 @@ public: ...@@ -182,7 +183,7 @@ public:
// max_depth < 0 means not limit // max_depth < 0 means not limit
int max_depth = -1; int max_depth = -1;
int top_k = 20; int top_k = 20;
void Set(const std::unordered_map<std::string, std::string>& params) override; LIGHTGBM_EXPORT void Set(const std::unordered_map<std::string, std::string>& params) override;
}; };
/*! \brief Config for Boosting */ /*! \brief Config for Boosting */
...@@ -207,7 +208,7 @@ public: ...@@ -207,7 +208,7 @@ public:
int drop_seed = 4; int drop_seed = 4;
std::string tree_learner_type = "serial"; std::string tree_learner_type = "serial";
TreeConfig tree_config; TreeConfig tree_config;
void Set(const std::unordered_map<std::string, std::string>& params) override; LIGHTGBM_EXPORT void Set(const std::unordered_map<std::string, std::string>& params) override;
private: private:
void GetTreeLearnerType(const std::unordered_map<std::string, void GetTreeLearnerType(const std::unordered_map<std::string,
std::string>& params); std::string>& params);
...@@ -220,7 +221,7 @@ public: ...@@ -220,7 +221,7 @@ public:
int local_listen_port = 12400; int local_listen_port = 12400;
int time_out = 120; // in minutes int time_out = 120; // in minutes
std::string machine_list_filename = ""; std::string machine_list_filename = "";
void Set(const std::unordered_map<std::string, std::string>& params) override; LIGHTGBM_EXPORT void Set(const std::unordered_map<std::string, std::string>& params) override;
}; };
...@@ -241,7 +242,7 @@ public: ...@@ -241,7 +242,7 @@ public:
std::vector<std::string> metric_types; std::vector<std::string> metric_types;
MetricConfig metric_config; MetricConfig metric_config;
void Set(const std::unordered_map<std::string, std::string>& params) override; LIGHTGBM_EXPORT void Set(const std::unordered_map<std::string, std::string>& params) override;
private: private:
void GetBoostingType(const std::unordered_map<std::string, std::string>& params); void GetBoostingType(const std::unordered_map<std::string, std::string>& params);
......
include/LightGBM/dataset.h
View file @ 46818ac6
...@@ -280,14 +280,14 @@ class Dataset { ...@@ -280,14 +280,14 @@ class Dataset {
public: public:
friend DatasetLoader; friend DatasetLoader;
Dataset(); LIGHTGBM_EXPORT Dataset();
Dataset(data_size_t num_data); LIGHTGBM_EXPORT Dataset(data_size_t num_data);
/*! \brief Destructor */ /*! \brief Destructor */
~Dataset(); LIGHTGBM_EXPORT ~Dataset();
bool CheckAlign(const Dataset& other) const { LIGHTGBM_EXPORT bool CheckAlign(const Dataset& other) const {
if (num_features_ != other.num_features_) { if (num_features_ != other.num_features_) {
return false; return false;
} }
...@@ -330,26 +330,26 @@ public: ...@@ -330,26 +330,26 @@ public:
Dataset* Subset(const data_size_t* used_indices, data_size_t num_used_indices, bool is_enable_sparse) const; Dataset* Subset(const data_size_t* used_indices, data_size_t num_used_indices, bool is_enable_sparse) const;
void FinishLoad(); LIGHTGBM_EXPORT void FinishLoad();
bool SetFloatField(const char* field_name, const float* field_data, data_size_t num_element); LIGHTGBM_EXPORT bool SetFloatField(const char* field_name, const float* field_data, data_size_t num_element);
bool SetDoubleField(const char* field_name, const double* field_data, data_size_t num_element); LIGHTGBM_EXPORT bool SetDoubleField(const char* field_name, const double* field_data, data_size_t num_element);
bool SetIntField(const char* field_name, const int* field_data, data_size_t num_element); LIGHTGBM_EXPORT bool SetIntField(const char* field_name, const int* field_data, data_size_t num_element);
bool GetFloatField(const char* field_name, data_size_t* out_len, const float** out_ptr); LIGHTGBM_EXPORT bool GetFloatField(const char* field_name, data_size_t* out_len, const float** out_ptr);
bool GetDoubleField(const char* field_name, data_size_t* out_len, const double** out_ptr); LIGHTGBM_EXPORT bool GetDoubleField(const char* field_name, data_size_t* out_len, const double** out_ptr);
bool GetIntField(const char* field_name, data_size_t* out_len, const int** out_ptr); LIGHTGBM_EXPORT bool GetIntField(const char* field_name, data_size_t* out_len, const int** out_ptr);
/*! /*!
* \brief Save current dataset into binary file, will save to "filename.bin" * \brief Save current dataset into binary file, will save to "filename.bin"
*/ */
void SaveBinaryFile(const char* bin_filename); LIGHTGBM_EXPORT void SaveBinaryFile(const char* bin_filename);
void CopyFeatureMapperFrom(const Dataset* dataset, bool is_enable_sparse); LIGHTGBM_EXPORT void CopyFeatureMapperFrom(const Dataset* dataset, bool is_enable_sparse);
/*! /*!
* \brief Get a feature pointer for specific index * \brief Get a feature pointer for specific index
......
include/LightGBM/dataset_loader.h
View file @ 46818ac6
...@@ -8,21 +8,21 @@ namespace LightGBM { ...@@ -8,21 +8,21 @@ namespace LightGBM {
class DatasetLoader { class DatasetLoader {
public: public:
DatasetLoader(const IOConfig& io_config, const PredictFunction& predict_fun, int num_class, const char* filename); LIGHTGBM_EXPORT DatasetLoader(const IOConfig& io_config, const PredictFunction& predict_fun, int num_class, const char* filename);
~DatasetLoader(); LIGHTGBM_EXPORT ~DatasetLoader();
Dataset* LoadFromFile(const char* filename, int rank, int num_machines); LIGHTGBM_EXPORT Dataset* LoadFromFile(const char* filename, int rank, int num_machines);
Dataset* LoadFromFile(const char* filename) { LIGHTGBM_EXPORT Dataset* LoadFromFile(const char* filename) {
return LoadFromFile(filename, 0, 1); return LoadFromFile(filename, 0, 1);
} }
Dataset* LoadFromFileAlignWithOtherDataset(const char* filename, const Dataset* train_data); LIGHTGBM_EXPORT Dataset* LoadFromFileAlignWithOtherDataset(const char* filename, const Dataset* train_data);
Dataset* LoadFromBinFile(const char* data_filename, const char* bin_filename, int rank, int num_machines); LIGHTGBM_EXPORT Dataset* LoadFromBinFile(const char* data_filename, const char* bin_filename, int rank, int num_machines);
Dataset* CostructFromSampleData(std::vector<std::vector<double>>& sample_values, size_t total_sample_size, data_size_t num_data); LIGHTGBM_EXPORT Dataset* CostructFromSampleData(std::vector<std::vector<double>>& sample_values, size_t total_sample_size, data_size_t num_data);
/*! \brief Disable copy */ /*! \brief Disable copy */
DatasetLoader& operator=(const DatasetLoader&) = delete; DatasetLoader& operator=(const DatasetLoader&) = delete;
......
include/LightGBM/export.h 0 → 100644
View file @ 46818ac6
#ifndef LIGHTGBM_EXPORT_H_
#define LIGHTGBM_EXPORT_H_
/** Macros for exporting symbols in MSVC/GCC/CLANG **/
#ifdef __cplusplus
#define LIGHTGBM_EXTERN_C extern "C"
#else
#define LIGHTGBM_EXTERN_C
#endif
#ifdef _MSC_VER
#define LIGHTGBM_EXPORT __declspec(dllexport)
#define LIGHTGBM_C_EXPORT LIGHTGBM_EXTERN_C __declspec(dllexport)
#else
#define LIGHTGBM_EXPORT
#define LIGHTGBM_C_EXPORT LIGHTGBM_EXTERN_C
#endif
#endif /** LIGHTGBM_EXPORT_H_ **/
include/LightGBM/metric.h
View file @ 46818ac6
...@@ -46,7 +46,7 @@ public: ...@@ -46,7 +46,7 @@ public:
* \param type Specific type of metric * \param type Specific type of metric
* \param config Config for metric * \param config Config for metric
*/ */
static Metric* CreateMetric(const std::string& type, const MetricConfig& config); LIGHTGBM_EXPORT static Metric* CreateMetric(const std::string& type, const MetricConfig& config);
}; };
......
include/LightGBM/objective_function.h
View file @ 46818ac6
...@@ -44,7 +44,7 @@ public: ...@@ -44,7 +44,7 @@ public:
* \param type Specific type of objective function * \param type Specific type of objective function
* \param config Config for objective function * \param config Config for objective function
*/ */
static ObjectiveFunction* CreateObjectiveFunction(const std::string& type, LIGHTGBM_EXPORT static ObjectiveFunction* CreateObjectiveFunction(const std::string& type,
const ObjectiveConfig& config); const ObjectiveConfig& config);
}; };
......
src/c_api.cpp
View file @ 46818ac6
...@@ -267,11 +267,11 @@ private: ...@@ -267,11 +267,11 @@ private:
// start of c_api functions // start of c_api functions
DllExport const char* LGBM_GetLastError() { LIGHTGBM_C_EXPORT const char* LGBM_GetLastError() {
return LastErrorMsg(); return LastErrorMsg();
} }
DllExport int LGBM_DatasetCreateFromFile(const char* filename, LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromFile(const char* filename,
const char* parameters, const char* parameters,
const DatasetHandle reference, const DatasetHandle reference,
DatasetHandle* out) { DatasetHandle* out) {
...@@ -289,7 +289,7 @@ DllExport int LGBM_DatasetCreateFromFile(const char* filename, ...@@ -289,7 +289,7 @@ DllExport int LGBM_DatasetCreateFromFile(const char* filename,
API_END(); API_END();
} }
DllExport int LGBM_DatasetCreateFromMat(const void* data, LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromMat(const void* data,
int data_type, int data_type,
int32_t nrow, int32_t nrow,
int32_t ncol, int32_t ncol,
...@@ -338,7 +338,7 @@ DllExport int LGBM_DatasetCreateFromMat(const void* data, ...@@ -338,7 +338,7 @@ DllExport int LGBM_DatasetCreateFromMat(const void* data,
API_END(); API_END();
} }
DllExport int LGBM_DatasetCreateFromCSR(const void* indptr, LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromCSR(const void* indptr,
int indptr_type, int indptr_type,
const int32_t* indices, const int32_t* indices,
const void* data, const void* data,
...@@ -400,7 +400,7 @@ DllExport int LGBM_DatasetCreateFromCSR(const void* indptr, ...@@ -400,7 +400,7 @@ DllExport int LGBM_DatasetCreateFromCSR(const void* indptr,
API_END(); API_END();
} }
DllExport int LGBM_DatasetCreateFromCSC(const void* col_ptr, LIGHTGBM_C_EXPORT int LGBM_DatasetCreateFromCSC(const void* col_ptr,
int col_ptr_type, int col_ptr_type,
const int32_t* indices, const int32_t* indices,
const void* data, const void* data,
...@@ -462,7 +462,7 @@ DllExport int LGBM_DatasetCreateFromCSC(const void* col_ptr, ...@@ -462,7 +462,7 @@ DllExport int LGBM_DatasetCreateFromCSC(const void* col_ptr,
API_END(); API_END();
} }
DllExport int LGBM_DatasetGetSubset( LIGHTGBM_C_EXPORT int LGBM_DatasetGetSubset(
const DatasetHandle handle, const DatasetHandle handle,
const int32_t* used_row_indices, const int32_t* used_row_indices,
int32_t num_used_row_indices, int32_t num_used_row_indices,
...@@ -482,7 +482,7 @@ DllExport int LGBM_DatasetGetSubset( ...@@ -482,7 +482,7 @@ DllExport int LGBM_DatasetGetSubset(
API_END(); API_END();
} }
DllExport int LGBM_DatasetSetFeatureNames( LIGHTGBM_C_EXPORT int LGBM_DatasetSetFeatureNames(
DatasetHandle handle, DatasetHandle handle,
const char** feature_names, const char** feature_names,
int num_feature_names) { int num_feature_names) {
...@@ -496,7 +496,7 @@ DllExport int LGBM_DatasetSetFeatureNames( ...@@ -496,7 +496,7 @@ DllExport int LGBM_DatasetSetFeatureNames(
API_END(); API_END();
} }
DllExport int LGBM_DatasetGetFeatureNames( LIGHTGBM_C_EXPORT int LGBM_DatasetGetFeatureNames(
DatasetHandle handle, DatasetHandle handle,
char** feature_names, char** feature_names,
int* num_feature_names) { int* num_feature_names) {
...@@ -510,13 +510,13 @@ DllExport int LGBM_DatasetGetFeatureNames( ...@@ -510,13 +510,13 @@ DllExport int LGBM_DatasetGetFeatureNames(
API_END(); API_END();
} }
DllExport int LGBM_DatasetFree(DatasetHandle handle) { LIGHTGBM_C_EXPORT int LGBM_DatasetFree(DatasetHandle handle) {
API_BEGIN(); API_BEGIN();
delete reinterpret_cast<Dataset*>(handle); delete reinterpret_cast<Dataset*>(handle);
API_END(); API_END();
} }
DllExport int LGBM_DatasetSaveBinary(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetSaveBinary(DatasetHandle handle,
const char* filename) { const char* filename) {
API_BEGIN(); API_BEGIN();
auto dataset = reinterpret_cast<Dataset*>(handle); auto dataset = reinterpret_cast<Dataset*>(handle);
...@@ -524,7 +524,7 @@ DllExport int LGBM_DatasetSaveBinary(DatasetHandle handle, ...@@ -524,7 +524,7 @@ DllExport int LGBM_DatasetSaveBinary(DatasetHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_DatasetSetField(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetSetField(DatasetHandle handle,
const char* field_name, const char* field_name,
const void* field_data, const void* field_data,
int num_element, int num_element,
...@@ -543,7 +543,7 @@ DllExport int LGBM_DatasetSetField(DatasetHandle handle, ...@@ -543,7 +543,7 @@ DllExport int LGBM_DatasetSetField(DatasetHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_DatasetGetField(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetGetField(DatasetHandle handle,
const char* field_name, const char* field_name,
int* out_len, int* out_len,
const void** out_ptr, const void** out_ptr,
...@@ -566,7 +566,7 @@ DllExport int LGBM_DatasetGetField(DatasetHandle handle, ...@@ -566,7 +566,7 @@ DllExport int LGBM_DatasetGetField(DatasetHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_DatasetGetNumData(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetGetNumData(DatasetHandle handle,
int* out) { int* out) {
API_BEGIN(); API_BEGIN();
auto dataset = reinterpret_cast<Dataset*>(handle); auto dataset = reinterpret_cast<Dataset*>(handle);
...@@ -574,7 +574,7 @@ DllExport int LGBM_DatasetGetNumData(DatasetHandle handle, ...@@ -574,7 +574,7 @@ DllExport int LGBM_DatasetGetNumData(DatasetHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_DatasetGetNumFeature(DatasetHandle handle, LIGHTGBM_C_EXPORT int LGBM_DatasetGetNumFeature(DatasetHandle handle,
int* out) { int* out) {
API_BEGIN(); API_BEGIN();
auto dataset = reinterpret_cast<Dataset*>(handle); auto dataset = reinterpret_cast<Dataset*>(handle);
...@@ -584,7 +584,7 @@ DllExport int LGBM_DatasetGetNumFeature(DatasetHandle handle, ...@@ -584,7 +584,7 @@ DllExport int LGBM_DatasetGetNumFeature(DatasetHandle handle,
// ---- start of booster // ---- start of booster
DllExport int LGBM_BoosterCreate(const DatasetHandle train_data, LIGHTGBM_C_EXPORT int LGBM_BoosterCreate(const DatasetHandle train_data,
const char* parameters, const char* parameters,
BoosterHandle* out) { BoosterHandle* out) {
API_BEGIN(); API_BEGIN();
...@@ -594,7 +594,7 @@ DllExport int LGBM_BoosterCreate(const DatasetHandle train_data, ...@@ -594,7 +594,7 @@ DllExport int LGBM_BoosterCreate(const DatasetHandle train_data,
API_END(); API_END();
} }
DllExport int LGBM_BoosterCreateFromModelfile( LIGHTGBM_C_EXPORT int LGBM_BoosterCreateFromModelfile(
const char* filename, const char* filename,
int* out_num_iterations, int* out_num_iterations,
BoosterHandle* out) { BoosterHandle* out) {
...@@ -605,13 +605,13 @@ DllExport int LGBM_BoosterCreateFromModelfile( ...@@ -605,13 +605,13 @@ DllExport int LGBM_BoosterCreateFromModelfile(
API_END(); API_END();
} }
DllExport int LGBM_BoosterFree(BoosterHandle handle) { LIGHTGBM_C_EXPORT int LGBM_BoosterFree(BoosterHandle handle) {
API_BEGIN(); API_BEGIN();
delete reinterpret_cast<Booster*>(handle); delete reinterpret_cast<Booster*>(handle);
API_END(); API_END();
} }
DllExport int LGBM_BoosterMerge(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterMerge(BoosterHandle handle,
BoosterHandle other_handle) { BoosterHandle other_handle) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
...@@ -620,7 +620,7 @@ DllExport int LGBM_BoosterMerge(BoosterHandle handle, ...@@ -620,7 +620,7 @@ DllExport int LGBM_BoosterMerge(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterAddValidData(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterAddValidData(BoosterHandle handle,
const DatasetHandle valid_data) { const DatasetHandle valid_data) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
...@@ -629,7 +629,7 @@ DllExport int LGBM_BoosterAddValidData(BoosterHandle handle, ...@@ -629,7 +629,7 @@ DllExport int LGBM_BoosterAddValidData(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterResetTrainingData(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterResetTrainingData(BoosterHandle handle,
const DatasetHandle train_data) { const DatasetHandle train_data) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
...@@ -638,21 +638,21 @@ DllExport int LGBM_BoosterResetTrainingData(BoosterHandle handle, ...@@ -638,21 +638,21 @@ DllExport int LGBM_BoosterResetTrainingData(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterResetParameter(BoosterHandle handle, const char* parameters) { LIGHTGBM_C_EXPORT int LGBM_BoosterResetParameter(BoosterHandle handle, const char* parameters) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
ref_booster->ResetConfig(parameters); ref_booster->ResetConfig(parameters);
API_END(); API_END();
} }
DllExport int LGBM_BoosterGetNumClasses(BoosterHandle handle, int* out_len) { LIGHTGBM_C_EXPORT int LGBM_BoosterGetNumClasses(BoosterHandle handle, int* out_len) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
*out_len = ref_booster->GetBoosting()->NumberOfClasses(); *out_len = ref_booster->GetBoosting()->NumberOfClasses();
API_END(); API_END();
} }
DllExport int LGBM_BoosterUpdateOneIter(BoosterHandle handle, int* is_finished) { LIGHTGBM_C_EXPORT int LGBM_BoosterUpdateOneIter(BoosterHandle handle, int* is_finished) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
if (ref_booster->TrainOneIter()) { if (ref_booster->TrainOneIter()) {
...@@ -663,7 +663,7 @@ DllExport int LGBM_BoosterUpdateOneIter(BoosterHandle handle, int* is_finished) ...@@ -663,7 +663,7 @@ DllExport int LGBM_BoosterUpdateOneIter(BoosterHandle handle, int* is_finished)
API_END(); API_END();
} }
DllExport int LGBM_BoosterUpdateOneIterCustom(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterUpdateOneIterCustom(BoosterHandle handle,
const float* grad, const float* grad,
const float* hess, const float* hess,
int* is_finished) { int* is_finished) {
...@@ -677,35 +677,35 @@ DllExport int LGBM_BoosterUpdateOneIterCustom(BoosterHandle handle, ...@@ -677,35 +677,35 @@ DllExport int LGBM_BoosterUpdateOneIterCustom(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterRollbackOneIter(BoosterHandle handle) { LIGHTGBM_C_EXPORT int LGBM_BoosterRollbackOneIter(BoosterHandle handle) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
ref_booster->RollbackOneIter(); ref_booster->RollbackOneIter();
API_END(); API_END();
} }
DllExport int LGBM_BoosterGetCurrentIteration(BoosterHandle handle, int* out_iteration) { LIGHTGBM_C_EXPORT int LGBM_BoosterGetCurrentIteration(BoosterHandle handle, int* out_iteration) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
*out_iteration = ref_booster->GetBoosting()->GetCurrentIteration(); *out_iteration = ref_booster->GetBoosting()->GetCurrentIteration();
API_END(); API_END();
} }
DllExport int LGBM_BoosterGetEvalCounts(BoosterHandle handle, int* out_len) { LIGHTGBM_C_EXPORT int LGBM_BoosterGetEvalCounts(BoosterHandle handle, int* out_len) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
*out_len = ref_booster->GetEvalCounts(); *out_len = ref_booster->GetEvalCounts();
API_END(); API_END();
} }
DllExport int LGBM_BoosterGetEvalNames(BoosterHandle handle, int* out_len, char** out_strs) { LIGHTGBM_C_EXPORT int LGBM_BoosterGetEvalNames(BoosterHandle handle, int* out_len, char** out_strs) {
API_BEGIN(); API_BEGIN();
Booster* ref_booster = reinterpret_cast<Booster*>(handle); Booster* ref_booster = reinterpret_cast<Booster*>(handle);
*out_len = ref_booster->GetEvalNames(out_strs); *out_len = ref_booster->GetEvalNames(out_strs);
API_END(); API_END();
} }
DllExport int LGBM_BoosterGetEval(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterGetEval(BoosterHandle handle,
int data_idx, int data_idx,
int* out_len, int* out_len,
double* out_results) { double* out_results) {
...@@ -720,7 +720,7 @@ DllExport int LGBM_BoosterGetEval(BoosterHandle handle, ...@@ -720,7 +720,7 @@ DllExport int LGBM_BoosterGetEval(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterGetNumPredict(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterGetNumPredict(BoosterHandle handle,
int data_idx, int data_idx,
int64_t* out_len) { int64_t* out_len) {
API_BEGIN(); API_BEGIN();
...@@ -729,7 +729,7 @@ DllExport int LGBM_BoosterGetNumPredict(BoosterHandle handle, ...@@ -729,7 +729,7 @@ DllExport int LGBM_BoosterGetNumPredict(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterGetPredict(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterGetPredict(BoosterHandle handle,
int data_idx, int data_idx,
int64_t* out_len, int64_t* out_len,
double* out_result) { double* out_result) {
...@@ -739,7 +739,7 @@ DllExport int LGBM_BoosterGetPredict(BoosterHandle handle, ...@@ -739,7 +739,7 @@ DllExport int LGBM_BoosterGetPredict(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterPredictForFile(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterPredictForFile(BoosterHandle handle,
const char* data_filename, const char* data_filename,
int data_has_header, int data_has_header,
int predict_type, int predict_type,
...@@ -766,7 +766,7 @@ int64_t GetNumPredOneRow(const Booster* ref_booster, int predict_type, int64_t n ...@@ -766,7 +766,7 @@ int64_t GetNumPredOneRow(const Booster* ref_booster, int predict_type, int64_t n
return num_preb_in_one_row; return num_preb_in_one_row;
} }
DllExport int LGBM_BoosterCalcNumPredict(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterCalcNumPredict(BoosterHandle handle,
int num_row, int num_row,
int predict_type, int predict_type,
int num_iteration, int num_iteration,
...@@ -777,7 +777,7 @@ DllExport int LGBM_BoosterCalcNumPredict(BoosterHandle handle, ...@@ -777,7 +777,7 @@ DllExport int LGBM_BoosterCalcNumPredict(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterPredictForCSR(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterPredictForCSR(BoosterHandle handle,
const void* indptr, const void* indptr,
int indptr_type, int indptr_type,
const int32_t* indices, const int32_t* indices,
...@@ -808,7 +808,7 @@ DllExport int LGBM_BoosterPredictForCSR(BoosterHandle handle, ...@@ -808,7 +808,7 @@ DllExport int LGBM_BoosterPredictForCSR(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterPredictForCSC(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterPredictForCSC(BoosterHandle handle,
const void* col_ptr, const void* col_ptr,
int col_ptr_type, int col_ptr_type,
const int32_t* indices, const int32_t* indices,
...@@ -853,7 +853,7 @@ DllExport int LGBM_BoosterPredictForCSC(BoosterHandle handle, ...@@ -853,7 +853,7 @@ DllExport int LGBM_BoosterPredictForCSC(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterPredictForMat(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterPredictForMat(BoosterHandle handle,
const void* data, const void* data,
int data_type, int data_type,
int32_t nrow, int32_t nrow,
...@@ -880,7 +880,7 @@ DllExport int LGBM_BoosterPredictForMat(BoosterHandle handle, ...@@ -880,7 +880,7 @@ DllExport int LGBM_BoosterPredictForMat(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterSaveModel(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterSaveModel(BoosterHandle handle,
int num_iteration, int num_iteration,
const char* filename) { const char* filename) {
API_BEGIN(); API_BEGIN();
...@@ -889,7 +889,7 @@ DllExport int LGBM_BoosterSaveModel(BoosterHandle handle, ...@@ -889,7 +889,7 @@ DllExport int LGBM_BoosterSaveModel(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterDumpModel(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterDumpModel(BoosterHandle handle,
int num_iteration, int num_iteration,
int buffer_len, int buffer_len,
int* out_len, int* out_len,
...@@ -904,7 +904,7 @@ DllExport int LGBM_BoosterDumpModel(BoosterHandle handle, ...@@ -904,7 +904,7 @@ DllExport int LGBM_BoosterDumpModel(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterGetLeafValue(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterGetLeafValue(BoosterHandle handle,
int tree_idx, int tree_idx,
int leaf_idx, int leaf_idx,
double* out_val) { double* out_val) {
...@@ -914,7 +914,7 @@ DllExport int LGBM_BoosterGetLeafValue(BoosterHandle handle, ...@@ -914,7 +914,7 @@ DllExport int LGBM_BoosterGetLeafValue(BoosterHandle handle,
API_END(); API_END();
} }
DllExport int LGBM_BoosterSetLeafValue(BoosterHandle handle, LIGHTGBM_C_EXPORT int LGBM_BoosterSetLeafValue(BoosterHandle handle,
int tree_idx, int tree_idx,
int leaf_idx, int leaf_idx,
double val) { double val) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment