Skip to content

Commit 3a35a82

Browse files
najeeb-kazmicodemzs
authored andcommitted
Change default EvaluationMetric for LightGbm trainers to conform to default metric in standalone LightGbm (#3859)
1 parent 4b0fa53 commit 3a35a82

9 files changed

+15
-15
lines changed

src/Microsoft.ML.LightGbm/LightGbmBinaryTrainer.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ public enum EvaluateMetricType
162162
[Argument(ArgumentType.AtMostOnce,
163163
HelpText = "Evaluation metrics.",
164164
ShortName = "em")]
165-
public EvaluateMetricType EvaluationMetric = EvaluateMetricType.Logloss;
165+
public EvaluateMetricType EvaluationMetric = EvaluateMetricType.Default;
166166

167167
static Options()
168168
{

src/Microsoft.ML.LightGbm/LightGbmMulticlassTrainer.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ public enum EvaluateMetricType
9595
[Argument(ArgumentType.AtMostOnce,
9696
HelpText = "Evaluation metrics.",
9797
ShortName = "em")]
98-
public EvaluateMetricType EvaluationMetric = EvaluateMetricType.Error;
98+
public EvaluateMetricType EvaluationMetric = EvaluateMetricType.Default;
9999

100100
static Options()
101101
{

src/Microsoft.ML.LightGbm/LightGbmRankingTrainer.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ public enum EvaluateMetricType
143143
[Argument(ArgumentType.AtMostOnce,
144144
HelpText = "Evaluation metrics.",
145145
ShortName = "em")]
146-
public EvaluateMetricType EvaluationMetric = EvaluateMetricType.NormalizedDiscountedCumulativeGain;
146+
public EvaluateMetricType EvaluationMetric = EvaluateMetricType.Default;
147147

148148
static Options()
149149
{

src/Microsoft.ML.LightGbm/LightGbmRegressionTrainer.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ public enum EvaluateMetricType
133133
[Argument(ArgumentType.AtMostOnce,
134134
HelpText = "Evaluation metrics.",
135135
ShortName = "em")]
136-
public EvaluateMetricType EvaluationMetric = EvaluateMetricType.RootMeanSquaredError;
136+
public EvaluateMetricType EvaluationMetric = EvaluateMetricType.Default;
137137

138138
static Options()
139139
{

test/BaselineOutput/Common/EntryPoints/core_manifest.json

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11535,7 +11535,7 @@
1153511535
"Required": false,
1153611536
"SortOrder": 150.0,
1153711537
"IsNullable": false,
11538-
"Default": "Logloss"
11538+
"Default": "Default"
1153911539
},
1154011540
{
1154111541
"Name": "MaximumBinCountPerFeature",
@@ -12020,7 +12020,7 @@
1202012020
"Required": false,
1202112021
"SortOrder": 150.0,
1202212022
"IsNullable": false,
12023-
"Default": "Error"
12023+
"Default": "Default"
1202412024
},
1202512025
{
1202612026
"Name": "MaximumBinCountPerFeature",
@@ -12517,7 +12517,7 @@
1251712517
"Required": false,
1251812518
"SortOrder": 150.0,
1251912519
"IsNullable": false,
12520-
"Default": "NormalizedDiscountedCumulativeGain"
12520+
"Default": "Default"
1252112521
},
1252212522
{
1252312523
"Name": "MaximumBinCountPerFeature",
@@ -12975,7 +12975,7 @@
1297512975
"Required": false,
1297612976
"SortOrder": 150.0,
1297712977
"IsNullable": false,
12978-
"Default": "RootMeanSquaredError"
12978+
"Default": "Default"
1297912979
},
1298012980
{
1298112981
"Name": "MaximumBinCountPerFeature",

test/BaselineOutput/Common/LightGBMR/LightGBMReg-CV-generatedRegressionDataset-out.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,10 +35,10 @@ Virtual memory usage(MB): %Number%
3535
[1] 'Loading data for LightGBM' started.
3636
[1] 'Loading data for LightGBM' finished in %Time%.
3737
[2] 'Training with LightGBM' started.
38-
[2] (%Time%) Iteration: 50 Training-rmse: 6.09160118577349
38+
[2] (%Time%) Iteration: 50 Training-: 37.107605006517
3939
[2] 'Training with LightGBM' finished in %Time%.
4040
[3] 'Loading data for LightGBM #2' started.
4141
[3] 'Loading data for LightGBM #2' finished in %Time%.
4242
[4] 'Training with LightGBM #2' started.
43-
[4] (%Time%) Iteration: 50 Training-rmse: 5.26343689176522
43+
[4] (%Time%) Iteration: 50 Training-: 27.7037679135951
4444
[4] 'Training with LightGBM #2' finished in %Time%.

test/BaselineOutput/Common/LightGBMR/LightGBMReg-TrainTest-generatedRegressionDataset-out.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ Virtual memory usage(MB): %Number%
2626
[1] 'Loading data for LightGBM' started.
2727
[1] 'Loading data for LightGBM' finished in %Time%.
2828
[2] 'Training with LightGBM' started.
29-
[2] (%Time%) Iteration: 50 Training-rmse: 5.10533343749577
29+
[2] (%Time%) Iteration: 50 Training-: 26.0644295080124
3030
[2] 'Training with LightGBM' finished in %Time%.
3131
[3] 'Saving model' started.
3232
[3] 'Saving model' finished in %Time%.
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
LightGBMR
2-
L1(avg) L2(avg) RMS(avg) Loss-fn(avg) R Squared /iter /lr /nl /mil /v /nt Learner Name Train Dataset Test Dataset Results File Run Time Physical Memory Virtual Memory Command Line Settings
3-
26.59978 1393.326 37.32081 1393.326 0.923402 50 0.2 20 10 + 1 LightGBMR %Data% %Output% 99 0 0 maml.exe CV tr=LightGBMR{nt=1 iter=50 em=RootMeanSquaredError v=+ lr=0.2 mil=10 nl=20} threads=- dout=%Output% loader=Text{col=Label:R4:11 col=Features:R4:0-10 sep=; header+} data=%Data% seed=1 /iter:50;/lr:0.2;/nl:20;/mil:10;/v:+;/nt:1
2+
L1(avg) L2(avg) RMS(avg) Loss-fn(avg) R Squared /em /iter /lr /nl /mil /v /nt Learner Name Train Dataset Test Dataset Results File Run Time Physical Memory Virtual Memory Command Line Settings
3+
26.59978 1393.326 37.32081 1393.326 0.923402 RootMeanSquaredError 50 0.2 20 10 + 1 LightGBMR %Data% %Output% 99 0 0 maml.exe CV tr=LightGBMR{nt=1 iter=50 em=RootMeanSquaredError v=+ lr=0.2 mil=10 nl=20} threads=- dout=%Output% loader=Text{col=Label:R4:11 col=Features:R4:0-10 sep=; header+} data=%Data% seed=1 /em:RootMeanSquaredError;/iter:50;/lr:0.2;/nl:20;/mil:10;/v:+;/nt:1
44

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
LightGBMR
2-
L1(avg) L2(avg) RMS(avg) Loss-fn(avg) R Squared /iter /lr /nl /mil /v /nt Learner Name Train Dataset Test Dataset Results File Run Time Physical Memory Virtual Memory Command Line Settings
3-
3.428896 25.23601 5.023546 25.23601 0.998616 50 0.2 20 10 + 1 LightGBMR %Data% %Data% %Output% 99 0 0 maml.exe TrainTest test=%Data% tr=LightGBMR{nt=1 iter=50 em=RootMeanSquaredError v=+ lr=0.2 mil=10 nl=20} dout=%Output% loader=Text{col=Label:R4:11 col=Features:R4:0-10 sep=; header+} data=%Data% out=%Output% seed=1 /iter:50;/lr:0.2;/nl:20;/mil:10;/v:+;/nt:1
2+
L1(avg) L2(avg) RMS(avg) Loss-fn(avg) R Squared /em /iter /lr /nl /mil /v /nt Learner Name Train Dataset Test Dataset Results File Run Time Physical Memory Virtual Memory Command Line Settings
3+
3.428896 25.23601 5.023546 25.23601 0.998616 RootMeanSquaredError 50 0.2 20 10 + 1 LightGBMR %Data% %Data% %Output% 99 0 0 maml.exe TrainTest test=%Data% tr=LightGBMR{nt=1 iter=50 em=RootMeanSquaredError v=+ lr=0.2 mil=10 nl=20} dout=%Output% loader=Text{col=Label:R4:11 col=Features:R4:0-10 sep=; header+} data=%Data% out=%Output% seed=1 /em:RootMeanSquaredError;/iter:50;/lr:0.2;/nl:20;/mil:10;/v:+;/nt:1
44

0 commit comments

Comments
 (0)