Skip to content

Commit 2dff87b

Browse files
fix bug
1 parent be2dbcb commit 2dff87b

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

src/Microsoft.ML.AutoML/API/AutoCatalog.cs

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,7 @@ public SweepablePipeline BinaryClassification(string labelColumnName = DefaultCo
381381
lbfgsLogisticRegressionOption.LabelColumnName = labelColumnName;
382382
lbfgsLogisticRegressionOption.FeatureColumnName = featureColumnName;
383383
lbfgsLogisticRegressionOption.ExampleWeightColumnName = exampleWeightColumnName;
384-
res.Add(SweepableEstimatorFactory.CreateLbfgsPoissonRegressionRegression(lbfgsLogisticRegressionOption, lbfgsLogisticRegressionSearchSpace ?? new SearchSpace<LbfgsOption>(lbfgsLogisticRegressionOption)));
384+
res.Add(SweepableEstimatorFactory.CreateLbfgsLogisticRegressionBinary(lbfgsLogisticRegressionOption, lbfgsLogisticRegressionSearchSpace ?? new SearchSpace<LbfgsOption>(lbfgsLogisticRegressionOption)));
385385
}
386386

387387
if (useSdcaLogisticRegression)
@@ -479,15 +479,6 @@ public SweepablePipeline MultiClassification(
479479
res.Add(SweepableEstimatorFactory.CreateLightGbmMulti(lgbmOption, lgbmSearchSpace ?? new SearchSpace<LgbmOption>(lgbmOption)));
480480
}
481481

482-
if (useLbfgsMaximumEntrophy)
483-
{
484-
lbfgsMaximumEntrophyOption = lbfgsMaximumEntrophyOption ?? new LbfgsOption();
485-
lbfgsMaximumEntrophyOption.LabelColumnName = labelColumnName;
486-
lbfgsMaximumEntrophyOption.FeatureColumnName = featureColumnName;
487-
lbfgsMaximumEntrophyOption.ExampleWeightColumnName = exampleWeightColumnName;
488-
res.Add(SweepableEstimatorFactory.CreateLbfgsMaximumEntropyMulti(lbfgsMaximumEntrophyOption, lbfgsMaximumEntrophySearchSpace ?? new SearchSpace<LbfgsOption>(lbfgsMaximumEntrophyOption)));
489-
}
490-
491482
if (useLbfgsLogisticRegression)
492483
{
493484
lbfgsLogisticRegressionOption = lbfgsLogisticRegressionOption ?? new LbfgsOption();
@@ -497,6 +488,16 @@ public SweepablePipeline MultiClassification(
497488
res.Add(SweepableEstimatorFactory.CreateLbfgsLogisticRegressionOva(lbfgsLogisticRegressionOption, lbfgsLogisticRegressionSearchSpace ?? new SearchSpace<LbfgsOption>(lbfgsLogisticRegressionOption)));
498489
}
499490

491+
492+
if (useLbfgsMaximumEntrophy)
493+
{
494+
lbfgsMaximumEntrophyOption = lbfgsMaximumEntrophyOption ?? new LbfgsOption();
495+
lbfgsMaximumEntrophyOption.LabelColumnName = labelColumnName;
496+
lbfgsMaximumEntrophyOption.FeatureColumnName = featureColumnName;
497+
lbfgsMaximumEntrophyOption.ExampleWeightColumnName = exampleWeightColumnName;
498+
res.Add(SweepableEstimatorFactory.CreateLbfgsMaximumEntropyMulti(lbfgsMaximumEntrophyOption, lbfgsMaximumEntrophySearchSpace ?? new SearchSpace<LbfgsOption>(lbfgsMaximumEntrophyOption)));
499+
}
500+
500501
if (useSdcaMaximumEntrophy)
501502
{
502503
sdcaMaximumEntrophyOption = sdcaMaximumEntrophyOption ?? new SdcaOption();

0 commit comments

Comments
 (0)