Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tianlh
LightGBM-DCU
Commits
9006d3f2
Commit
9006d3f2
authored
Nov 24, 2016
by
Guolin Ke
Browse files
merge from master
parents
b0b0010a
65e711a2
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
17 additions
and
17 deletions
+17
-17
include/LightGBM/c_api.h
include/LightGBM/c_api.h
+17
-17
No files found.
include/LightGBM/c_api.h
View file @
9006d3f2
...
@@ -7,9 +7,9 @@
...
@@ -7,9 +7,9 @@
/*!
/*!
* To avoid type conversion on large data, most of our expose interface support both for float_32 and float_64.
* To avoid type conversion on large data, most of our expose interface support both for float_32 and float_64.
* Except following:
* Except following:
* 1. gradients and hessians.
* 1. gradients and hessians.
* 2. Get current score for training data and validation
* 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.
* The reason is because they are called frequently, the type-conversion on them maybe time cost.
*/
*/
#ifdef __cplusplus
#ifdef __cplusplus
...
@@ -362,10 +362,10 @@ DllExport int LGBM_BoosterGetPredict(BoosterHandle handle,
...
@@ -362,10 +362,10 @@ DllExport int LGBM_BoosterGetPredict(BoosterHandle handle,
* \param data_filename filename of data file
* \param data_filename filename of data file
* \param data_has_header data file has header or not
* \param data_has_header data file has header or not
* \param predict_type
* \param predict_type
* 0:
raw score
* 0:
normal, with transform (if needed)
* 1:
with transform(if needed)
* 1:
raw score
* 2:leaf index
* 2:leaf index
* \param num_iteration number of iteration for prediction
* \param num_iteration number of iteration for prediction
, < 0 means no limit
* \param result_filename filename of result file
* \param result_filename filename of result file
* \return 0 when success, -1 when failure happens
* \return 0 when success, -1 when failure happens
*/
*/
...
@@ -380,7 +380,7 @@ DllExport int LGBM_BoosterPredictForFile(BoosterHandle handle,
...
@@ -380,7 +380,7 @@ DllExport int LGBM_BoosterPredictForFile(BoosterHandle handle,
* \brief make prediction for an new data set
* \brief make prediction for an new data set
* \param handle handle
* \param handle handle
* \param indptr pointer to row headers
* \param indptr pointer to row headers
* \param indptr_type
* \param indptr_type
* \param indices findex
* \param indices findex
* \param data fvalue
* \param data fvalue
* \param data_type
* \param data_type
...
@@ -388,10 +388,10 @@ DllExport int LGBM_BoosterPredictForFile(BoosterHandle handle,
...
@@ -388,10 +388,10 @@ DllExport int LGBM_BoosterPredictForFile(BoosterHandle handle,
* \param nelem number of nonzero elements in the matrix
* \param nelem number of nonzero elements in the matrix
* \param num_col number of columns; when it's set to 0, then guess from data
* \param num_col number of columns; when it's set to 0, then guess from data
* \param predict_type
* \param predict_type
* 0:
raw score
* 0:
normal, with transform (if needed)
* 1:
with transform(if needed)
* 1:
raw score
* 2:leaf index
* 2:leaf index
* \param num_iteration number of iteration for prediction
* \param num_iteration number of iteration for prediction
, < 0 means no limit
* \param out_len len of output result
* \param out_len len of output result
* \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 success, -1 when failure happens
* \return 0 when success, -1 when failure happens
...
@@ -419,10 +419,10 @@ DllExport int LGBM_BoosterPredictForCSR(BoosterHandle handle,
...
@@ -419,10 +419,10 @@ DllExport int LGBM_BoosterPredictForCSR(BoosterHandle handle,
* \param ncol number columns
* \param ncol number columns
* \param is_row_major 1 for row major, 0 for column major
* \param is_row_major 1 for row major, 0 for column major
* \param predict_type
* \param predict_type
* 0:
raw score
* 0:
normal, with transform (if needed)
* 1:
with transform(if needed)
* 1:
raw score
* 2:leaf index
* 2:leaf index
* \param num_iteration number of iteration for prediction
* \param num_iteration number of iteration for prediction
, < 0 means no limit
* \param out_len len of output result
* \param out_len len of output result
* \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 success, -1 when failure happens
* \return 0 when success, -1 when failure happens
...
@@ -441,7 +441,7 @@ DllExport int LGBM_BoosterPredictForMat(BoosterHandle handle,
...
@@ -441,7 +441,7 @@ DllExport int LGBM_BoosterPredictForMat(BoosterHandle handle,
/*!
/*!
* \brief save model into file
* \brief save model into file
* \param handle handle
* \param handle handle
* \param num_iteration
* \param num_iteration
, < 0 means no limit
* \param filename file name
* \param filename file name
* \return 0 when success, -1 when failure happens
* \return 0 when success, -1 when failure happens
*/
*/
...
@@ -460,14 +460,14 @@ std::function<std::vector<std::pair<int, double>>(int row_idx)>
...
@@ -460,14 +460,14 @@ std::function<std::vector<std::pair<int, double>>(int row_idx)>
RowPairFunctionFromDenseMatric
(
const
void
*
data
,
int
num_row
,
int
num_col
,
int
data_type
,
int
is_row_major
);
RowPairFunctionFromDenseMatric
(
const
void
*
data
,
int
num_row
,
int
num_col
,
int
data_type
,
int
is_row_major
);
std
::
function
<
std
::
vector
<
std
::
pair
<
int
,
double
>>
(
int
idx
)
>
std
::
function
<
std
::
vector
<
std
::
pair
<
int
,
double
>>
(
int
idx
)
>
RowFunctionFromCSR
(
const
void
*
indptr
,
int
indptr_type
,
const
int32_t
*
indices
,
RowFunctionFromCSR
(
const
void
*
indptr
,
int
indptr_type
,
const
int32_t
*
indices
,
const
void
*
data
,
int
data_type
,
int64_t
nindptr
,
int64_t
nelem
);
const
void
*
data
,
int
data_type
,
int64_t
nindptr
,
int64_t
nelem
);
std
::
function
<
std
::
vector
<
std
::
pair
<
int
,
double
>>
(
int
idx
)
>
std
::
function
<
std
::
vector
<
std
::
pair
<
int
,
double
>>
(
int
idx
)
>
ColumnFunctionFromCSC
(
const
void
*
col_ptr
,
int
col_ptr_type
,
const
int32_t
*
indices
,
ColumnFunctionFromCSC
(
const
void
*
col_ptr
,
int
col_ptr_type
,
const
int32_t
*
indices
,
const
void
*
data
,
int
data_type
,
int64_t
ncol_ptr
,
int64_t
nelem
);
const
void
*
data
,
int
data_type
,
int64_t
ncol_ptr
,
int64_t
nelem
);
std
::
vector
<
double
>
std
::
vector
<
double
>
SampleFromOneColumn
(
const
std
::
vector
<
std
::
pair
<
int
,
double
>>&
data
,
const
std
::
vector
<
int
>&
indices
);
SampleFromOneColumn
(
const
std
::
vector
<
std
::
pair
<
int
,
double
>>&
data
,
const
std
::
vector
<
int
>&
indices
);
...
@@ -494,6 +494,6 @@ inline int LGBM_APIHandleException(const std::string& ex) {
...
@@ -494,6 +494,6 @@ inline int LGBM_APIHandleException(const std::string& ex) {
catch(std::exception& ex) { return LGBM_APIHandleException(ex); } \
catch(std::exception& ex) { return LGBM_APIHandleException(ex); } \
catch(std::string& ex) { return LGBM_APIHandleException(ex); } \
catch(std::string& ex) { return LGBM_APIHandleException(ex); } \
catch(...) { return LGBM_APIHandleException("unknown exception"); } \
catch(...) { return LGBM_APIHandleException("unknown exception"); } \
return 0;
return 0;
#endif // LIGHTGBM_C_API_H_
#endif // LIGHTGBM_C_API_H_
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment