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
26b7e818
Commit
26b7e818
authored
Dec 23, 2016
by
Guolin Ke
Browse files
typo
parent
006391b9
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
43 additions
and
43 deletions
+43
-43
include/LightGBM/c_api.h
include/LightGBM/c_api.h
+23
-23
src/c_api.cpp
src/c_api.cpp
+20
-20
No files found.
include/LightGBM/c_api.h
View file @
26b7e818
...
...
@@ -26,7 +26,7 @@
#define DllExport DLL_EXTERN_C
#endif
typedef
void
*
Dat
e
setHandle
;
typedef
void
*
Dat
a
setHandle
;
typedef
void
*
BoosterHandle
;
#define C_API_DTYPE_FLOAT32 (0)
...
...
@@ -59,8 +59,8 @@ DllExport const char* LGBM_GetLastError();
*/
DllExport
int
LGBM_DatasetCreateFromFile
(
const
char
*
filename
,
const
char
*
parameters
,
const
Dat
e
setHandle
*
reference
,
Dat
e
setHandle
*
out
);
const
Dat
a
setHandle
*
reference
,
Dat
a
setHandle
*
out
);
/*!
* \brief create a dataset from CSR format
...
...
@@ -86,8 +86,8 @@ DllExport int LGBM_DatasetCreateFromCSR(const void* indptr,
int64_t
nelem
,
int64_t
num_col
,
const
char
*
parameters
,
const
Dat
e
setHandle
*
reference
,
Dat
e
setHandle
*
out
);
const
Dat
a
setHandle
*
reference
,
Dat
a
setHandle
*
out
);
/*!
* \brief create a dataset from CSC format
...
...
@@ -113,8 +113,8 @@ DllExport int LGBM_DatasetCreateFromCSC(const void* col_ptr,
int64_t
nelem
,
int64_t
num_row
,
const
char
*
parameters
,
const
Dat
e
setHandle
*
reference
,
Dat
e
setHandle
*
out
);
const
Dat
a
setHandle
*
reference
,
Dat
a
setHandle
*
out
);
/*!
* \brief create dataset from dense matrix
...
...
@@ -134,8 +134,8 @@ DllExport int LGBM_DatasetCreateFromMat(const void* data,
int32_t
ncol
,
int
is_row_major
,
const
char
*
parameters
,
const
Dat
e
setHandle
*
reference
,
Dat
e
setHandle
*
out
);
const
Dat
a
setHandle
*
reference
,
Dat
a
setHandle
*
out
);
/*!
* \brief Create subset of a data
...
...
@@ -147,11 +147,11 @@ DllExport int LGBM_DatasetCreateFromMat(const void* data,
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_DatasetGetSubset
(
const
Dat
e
setHandle
*
handle
,
const
Dat
a
setHandle
*
handle
,
const
int32_t
*
used_row_indices
,
int32_t
num_used_row_indices
,
const
char
*
parameters
,
Dat
e
setHandle
*
out
);
Dat
a
setHandle
*
out
);
/*!
* \brief save feature names to Dataset
...
...
@@ -161,7 +161,7 @@ DllExport int LGBM_DatasetGetSubset(
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_DatasetSetFeatureNames
(
Dat
e
setHandle
handle
,
Dat
a
setHandle
handle
,
const
char
**
feature_names
,
int64_t
num_feature_names
);
...
...
@@ -169,7 +169,7 @@ DllExport int LGBM_DatasetSetFeatureNames(
* \brief free space for dataset
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_DatasetFree
(
Dat
e
setHandle
handle
);
DllExport
int
LGBM_DatasetFree
(
Dat
a
setHandle
handle
);
/*!
* \brief save dateset to binary file
...
...
@@ -177,7 +177,7 @@ DllExport int LGBM_DatasetFree(DatesetHandle handle);
* \param filename file name
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_DatasetSaveBinary
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetSaveBinary
(
Dat
a
setHandle
handle
,
const
char
*
filename
);
/*!
...
...
@@ -191,7 +191,7 @@ DllExport int LGBM_DatasetSaveBinary(DatesetHandle handle,
* \param type C_API_DTYPE_FLOAT32 or C_API_DTYPE_INT32
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_DatasetSetField
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetSetField
(
Dat
a
setHandle
handle
,
const
char
*
field_name
,
const
void
*
field_data
,
int64_t
num_element
,
...
...
@@ -206,7 +206,7 @@ DllExport int LGBM_DatasetSetField(DatesetHandle handle,
* \param out_type C_API_DTYPE_FLOAT32 or C_API_DTYPE_INT32
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_DatasetGetField
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetGetField
(
Dat
a
setHandle
handle
,
const
char
*
field_name
,
int64_t
*
out_len
,
const
void
**
out_ptr
,
...
...
@@ -218,7 +218,7 @@ DllExport int LGBM_DatasetGetField(DatesetHandle handle,
* \param out The address to hold number of data
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_DatasetGetNumData
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetGetNumData
(
Dat
a
setHandle
handle
,
int64_t
*
out
);
/*!
...
...
@@ -227,7 +227,7 @@ DllExport int LGBM_DatasetGetNumData(DatesetHandle handle,
* \param out The output of number of features
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_DatasetGetNumFeature
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetGetNumFeature
(
Dat
a
setHandle
handle
,
int64_t
*
out
);
// --- start Booster interfaces
...
...
@@ -239,7 +239,7 @@ DllExport int LGBM_DatasetGetNumFeature(DatesetHandle handle,
* \prama out handle of created Booster
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_BoosterCreate
(
const
Dat
e
setHandle
train_data
,
DllExport
int
LGBM_BoosterCreate
(
const
Dat
a
setHandle
train_data
,
const
char
*
parameters
,
BoosterHandle
*
out
);
...
...
@@ -279,7 +279,7 @@ DllExport int LGBM_BoosterMerge(BoosterHandle handle,
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_BoosterAddValidData
(
BoosterHandle
handle
,
const
Dat
e
setHandle
valid_data
);
const
Dat
a
setHandle
valid_data
);
/*!
* \brief Reset training data for booster
...
...
@@ -288,7 +288,7 @@ DllExport int LGBM_BoosterAddValidData(BoosterHandle handle,
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_BoosterResetTrainingData
(
BoosterHandle
handle
,
const
Dat
e
setHandle
train_data
);
const
Dat
a
setHandle
train_data
);
/*!
* \brief Reset config for current booster
...
...
@@ -352,7 +352,7 @@ DllExport int LGBM_BoosterGetEvalCounts(BoosterHandle handle, int64_t* out_len);
/*!
* \brief Get Name of eval
* \param out_len total number of eval results
* \param out_strs names of eval result
* \param out_strs names of eval result
, need to pre-allocate memory before call this
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_BoosterGetEvalNames
(
BoosterHandle
handle
,
int64_t
*
out_len
,
char
**
out_strs
);
...
...
@@ -491,7 +491,7 @@ DllExport int LGBM_BoosterSaveModel(BoosterHandle handle,
* \param handle handle
* \param buffer_len string buffer length, if buffer_len < out_len, re-allocate buffer
* \param out_len actual output length
* \param out_str json format string of model
* \param out_str json format string of model
, need to pre-allocate memory before call this
* \return 0 when succeed, -1 when failure happens
*/
DllExport
int
LGBM_BoosterDumpModel
(
BoosterHandle
handle
,
...
...
src/c_api.cpp
View file @
26b7e818
...
...
@@ -239,8 +239,8 @@ DllExport const char* LGBM_GetLastError() {
DllExport
int
LGBM_DatasetCreateFromFile
(
const
char
*
filename
,
const
char
*
parameters
,
const
Dat
e
setHandle
*
reference
,
Dat
e
setHandle
*
out
)
{
const
Dat
a
setHandle
*
reference
,
Dat
a
setHandle
*
out
)
{
API_BEGIN
();
auto
param
=
ConfigBase
::
Str2Map
(
parameters
);
IOConfig
io_config
;
...
...
@@ -261,8 +261,8 @@ DllExport int LGBM_DatasetCreateFromMat(const void* data,
int32_t
ncol
,
int
is_row_major
,
const
char
*
parameters
,
const
Dat
e
setHandle
*
reference
,
Dat
e
setHandle
*
out
)
{
const
Dat
a
setHandle
*
reference
,
Dat
a
setHandle
*
out
)
{
API_BEGIN
();
auto
param
=
ConfigBase
::
Str2Map
(
parameters
);
IOConfig
io_config
;
...
...
@@ -313,8 +313,8 @@ DllExport int LGBM_DatasetCreateFromCSR(const void* indptr,
int64_t
nelem
,
int64_t
num_col
,
const
char
*
parameters
,
const
Dat
e
setHandle
*
reference
,
Dat
e
setHandle
*
out
)
{
const
Dat
a
setHandle
*
reference
,
Dat
a
setHandle
*
out
)
{
API_BEGIN
();
auto
param
=
ConfigBase
::
Str2Map
(
parameters
);
IOConfig
io_config
;
...
...
@@ -375,8 +375,8 @@ DllExport int LGBM_DatasetCreateFromCSC(const void* col_ptr,
int64_t
nelem
,
int64_t
num_row
,
const
char
*
parameters
,
const
Dat
e
setHandle
*
reference
,
Dat
e
setHandle
*
out
)
{
const
Dat
a
setHandle
*
reference
,
Dat
a
setHandle
*
out
)
{
API_BEGIN
();
auto
param
=
ConfigBase
::
Str2Map
(
parameters
);
IOConfig
io_config
;
...
...
@@ -417,11 +417,11 @@ DllExport int LGBM_DatasetCreateFromCSC(const void* col_ptr,
}
DllExport
int
LGBM_DatasetGetSubset
(
const
Dat
e
setHandle
*
handle
,
const
Dat
a
setHandle
*
handle
,
const
int32_t
*
used_row_indices
,
int32_t
num_used_row_indices
,
const
char
*
parameters
,
Dat
e
setHandle
*
out
)
{
Dat
a
setHandle
*
out
)
{
API_BEGIN
();
auto
param
=
ConfigBase
::
Str2Map
(
parameters
);
IOConfig
io_config
;
...
...
@@ -437,7 +437,7 @@ DllExport int LGBM_DatasetGetSubset(
}
DllExport
int
LGBM_DatasetSetFeatureNames
(
Dat
e
setHandle
handle
,
Dat
a
setHandle
handle
,
const
char
**
feature_names
,
int64_t
num_feature_names
)
{
API_BEGIN
();
...
...
@@ -450,13 +450,13 @@ DllExport int LGBM_DatasetSetFeatureNames(
API_END
();
}
DllExport
int
LGBM_DatasetFree
(
Dat
e
setHandle
handle
)
{
DllExport
int
LGBM_DatasetFree
(
Dat
a
setHandle
handle
)
{
API_BEGIN
();
delete
reinterpret_cast
<
Dataset
*>
(
handle
);
API_END
();
}
DllExport
int
LGBM_DatasetSaveBinary
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetSaveBinary
(
Dat
a
setHandle
handle
,
const
char
*
filename
)
{
API_BEGIN
();
auto
dataset
=
reinterpret_cast
<
Dataset
*>
(
handle
);
...
...
@@ -464,7 +464,7 @@ DllExport int LGBM_DatasetSaveBinary(DatesetHandle handle,
API_END
();
}
DllExport
int
LGBM_DatasetSetField
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetSetField
(
Dat
a
setHandle
handle
,
const
char
*
field_name
,
const
void
*
field_data
,
int64_t
num_element
,
...
...
@@ -481,7 +481,7 @@ DllExport int LGBM_DatasetSetField(DatesetHandle handle,
API_END
();
}
DllExport
int
LGBM_DatasetGetField
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetGetField
(
Dat
a
setHandle
handle
,
const
char
*
field_name
,
int64_t
*
out_len
,
const
void
**
out_ptr
,
...
...
@@ -501,7 +501,7 @@ DllExport int LGBM_DatasetGetField(DatesetHandle handle,
API_END
();
}
DllExport
int
LGBM_DatasetGetNumData
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetGetNumData
(
Dat
a
setHandle
handle
,
int64_t
*
out
)
{
API_BEGIN
();
auto
dataset
=
reinterpret_cast
<
Dataset
*>
(
handle
);
...
...
@@ -509,7 +509,7 @@ DllExport int LGBM_DatasetGetNumData(DatesetHandle handle,
API_END
();
}
DllExport
int
LGBM_DatasetGetNumFeature
(
Dat
e
setHandle
handle
,
DllExport
int
LGBM_DatasetGetNumFeature
(
Dat
a
setHandle
handle
,
int64_t
*
out
)
{
API_BEGIN
();
auto
dataset
=
reinterpret_cast
<
Dataset
*>
(
handle
);
...
...
@@ -520,7 +520,7 @@ DllExport int LGBM_DatasetGetNumFeature(DatesetHandle handle,
// ---- start of booster
DllExport
int
LGBM_BoosterCreate
(
const
Dat
e
setHandle
train_data
,
DllExport
int
LGBM_BoosterCreate
(
const
Dat
a
setHandle
train_data
,
const
char
*
parameters
,
BoosterHandle
*
out
)
{
API_BEGIN
();
...
...
@@ -558,7 +558,7 @@ DllExport int LGBM_BoosterMerge(BoosterHandle handle,
}
DllExport
int
LGBM_BoosterAddValidData
(
BoosterHandle
handle
,
const
Dat
e
setHandle
valid_data
)
{
const
Dat
a
setHandle
valid_data
)
{
API_BEGIN
();
Booster
*
ref_booster
=
reinterpret_cast
<
Booster
*>
(
handle
);
const
Dataset
*
p_dataset
=
reinterpret_cast
<
const
Dataset
*>
(
valid_data
);
...
...
@@ -567,7 +567,7 @@ DllExport int LGBM_BoosterAddValidData(BoosterHandle handle,
}
DllExport
int
LGBM_BoosterResetTrainingData
(
BoosterHandle
handle
,
const
Dat
e
setHandle
train_data
)
{
const
Dat
a
setHandle
train_data
)
{
API_BEGIN
();
Booster
*
ref_booster
=
reinterpret_cast
<
Booster
*>
(
handle
);
const
Dataset
*
p_dataset
=
reinterpret_cast
<
const
Dataset
*>
(
train_data
);
...
...
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