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
8e190f59
Commit
8e190f59
authored
Feb 27, 2017
by
zhangyafeikimi
Committed by
Guolin Ke
Mar 01, 2017
Browse files
fix typos and comments (#326)
parent
362f9ac4
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
8 additions
and
8 deletions
+8
-8
include/LightGBM/application.h
include/LightGBM/application.h
+4
-4
src/application/predictor.hpp
src/application/predictor.hpp
+3
-4
src/io/dataset.cpp
src/io/dataset.cpp
+1
-0
No files found.
include/LightGBM/application.h
View file @
8e190f59
...
@@ -19,8 +19,8 @@ class Metric;
...
@@ -19,8 +19,8 @@ class Metric;
* \brief The main entrance of LightGBM. this application has two tasks:
* \brief The main entrance of LightGBM. this application has two tasks:
* Train and Predict.
* Train and Predict.
* Train task will train a new model
* Train task will train a new model
* Predict task will predict
ing
the scores of test data using exsiting model,
* Predict task will predict the scores of test data using exsi
s
ting model,
* and sav
ing
the score to disk.
* and sav
e
the score to disk.
*/
*/
class
Application
{
class
Application
{
public:
public:
...
@@ -41,7 +41,7 @@ private:
...
@@ -41,7 +41,7 @@ private:
template
<
typename
T
>
template
<
typename
T
>
T
GlobalSyncUpByMin
(
T
&
local
);
T
GlobalSyncUpByMin
(
T
&
local
);
/*! \brief Load parametes from command line and config file*/
/*! \brief Load paramete
r
s from command line and config file*/
void
LoadParameters
(
int
argc
,
char
**
argv
);
void
LoadParameters
(
int
argc
,
char
**
argv
);
/*! \brief Load data, including training data and validation data*/
/*! \brief Load data, including training data and validation data*/
...
...
src/application/predictor.hpp
View file @
8e190f59
...
@@ -19,7 +19,7 @@
...
@@ -19,7 +19,7 @@
namespace
LightGBM
{
namespace
LightGBM
{
/*!
/*!
* \brief Used to predict
ion
data with input model
* \brief Used to predict data with input model
*/
*/
class
Predictor
{
class
Predictor
{
public:
public:
...
@@ -27,7 +27,7 @@ public:
...
@@ -27,7 +27,7 @@ public:
* \brief Constructor
* \brief Constructor
* \param boosting Input boosting model
* \param boosting Input boosting model
* \param is_raw_score True if need to predict result with raw score
* \param is_raw_score True if need to predict result with raw score
* \param predict_leaf_index True if output leaf index instead of prediction score
* \param
is_
predict_leaf_index True if output leaf index instead of prediction score
*/
*/
Predictor
(
const
Boosting
*
boosting
,
bool
is_raw_score
,
bool
is_predict_leaf_index
)
{
Predictor
(
const
Boosting
*
boosting
,
bool
is_raw_score
,
bool
is_predict_leaf_index
)
{
boosting_
=
boosting
;
boosting_
=
boosting
;
...
@@ -69,14 +69,13 @@ public:
...
@@ -69,14 +69,13 @@ public:
~
Predictor
()
{
~
Predictor
()
{
}
}
inline
const
PredictFunction
&
GetPredictFunction
()
{
inline
const
PredictFunction
&
GetPredictFunction
()
const
{
return
predict_fun_
;
return
predict_fun_
;
}
}
/*!
/*!
* \brief predicting on data, then saving result to disk
* \brief predicting on data, then saving result to disk
* \param data_filename Filename of data
* \param data_filename Filename of data
* \param has_label True if this data contains label
* \param result_filename Filename of output result
* \param result_filename Filename of output result
*/
*/
void
Predict
(
const
char
*
data_filename
,
const
char
*
result_filename
,
bool
has_header
)
{
void
Predict
(
const
char
*
data_filename
,
const
char
*
result_filename
,
bool
has_header
)
{
...
...
src/io/dataset.cpp
View file @
8e190f59
...
@@ -4,6 +4,7 @@
...
@@ -4,6 +4,7 @@
#include <LightGBM/utils/threading.h>
#include <LightGBM/utils/threading.h>
#include <LightGBM/utils/array_args.h>
#include <LightGBM/utils/array_args.h>
#include <chrono>
#include <cstdio>
#include <cstdio>
#include <unordered_map>
#include <unordered_map>
#include <limits>
#include <limits>
...
...
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