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
9f04f276
Commit
9f04f276
authored
Nov 12, 2016
by
Guolin Ke
Browse files
change dropping_seed to drop_seed
parent
47b7455d
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
3 deletions
+3
-3
include/LightGBM/config.h
include/LightGBM/config.h
+1
-1
src/boosting/dart.cpp
src/boosting/dart.cpp
+1
-1
src/io/config.cpp
src/io/config.cpp
+1
-1
No files found.
include/LightGBM/config.h
View file @
9f04f276
...
@@ -192,7 +192,7 @@ public:
...
@@ -192,7 +192,7 @@ public:
int
early_stopping_round
=
0
;
int
early_stopping_round
=
0
;
int
num_class
=
1
;
int
num_class
=
1
;
double
drop_rate
=
0.01
;
double
drop_rate
=
0.01
;
int
drop
ping
_seed
=
4
;
int
drop_seed
=
4
;
void
Set
(
const
std
::
unordered_map
<
std
::
string
,
std
::
string
>&
params
)
override
;
void
Set
(
const
std
::
unordered_map
<
std
::
string
,
std
::
string
>&
params
)
override
;
};
};
...
...
src/boosting/dart.cpp
View file @
9f04f276
...
@@ -29,7 +29,7 @@ void DART::Init(const BoostingConfig* config, const Dataset* train_data, const O
...
@@ -29,7 +29,7 @@ void DART::Init(const BoostingConfig* config, const Dataset* train_data, const O
gbdt_config_
=
dynamic_cast
<
const
GBDTConfig
*>
(
config
);
gbdt_config_
=
dynamic_cast
<
const
GBDTConfig
*>
(
config
);
drop_rate_
=
gbdt_config_
->
drop_rate
;
drop_rate_
=
gbdt_config_
->
drop_rate
;
shrinkage_rate_
=
1.0
;
shrinkage_rate_
=
1.0
;
random_for_drop_
=
Random
(
gbdt_config_
->
drop
ping
_seed
);
random_for_drop_
=
Random
(
gbdt_config_
->
drop_seed
);
}
}
bool
DART
::
TrainOneIter
(
const
score_t
*
gradient
,
const
score_t
*
hessian
,
bool
is_eval
)
{
bool
DART
::
TrainOneIter
(
const
score_t
*
gradient
,
const
score_t
*
hessian
,
bool
is_eval
)
{
...
...
src/io/config.cpp
View file @
9f04f276
...
@@ -298,7 +298,7 @@ void BoostingConfig::Set(const std::unordered_map<std::string, std::string>& par
...
@@ -298,7 +298,7 @@ void BoostingConfig::Set(const std::unordered_map<std::string, std::string>& par
CHECK
(
output_freq
>=
0
);
CHECK
(
output_freq
>=
0
);
GetBool
(
params
,
"is_training_metric"
,
&
is_provide_training_metric
);
GetBool
(
params
,
"is_training_metric"
,
&
is_provide_training_metric
);
GetInt
(
params
,
"num_class"
,
&
num_class
);
GetInt
(
params
,
"num_class"
,
&
num_class
);
GetInt
(
params
,
"drop
ping
_seed"
,
&
drop
ping
_seed
);
GetInt
(
params
,
"drop_seed"
,
&
drop_seed
);
GetDouble
(
params
,
"drop_rate"
,
&
drop_rate
);
GetDouble
(
params
,
"drop_rate"
,
&
drop_rate
);
CHECK
(
drop_rate
<=
1.0
&&
drop_rate
>=
0.0
);
CHECK
(
drop_rate
<=
1.0
&&
drop_rate
>=
0.0
);
}
}
...
...
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