Skip to content

Commit 8af0d21

Browse files
committed
Fix newly-added file
Remove empty line to trigger build
1 parent 7750434 commit 8af0d21

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

test/Microsoft.ML.Functional.Tests/Prediction.cs

+4-5
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@ public void ReconfigurablePrediction()
4444
// Create a training pipeline.
4545
var pipeline = mlContext.Transforms.Text.FeaturizeText("Features", "SentimentText")
4646
.AppendCacheCheckpoint(mlContext)
47-
.Append(mlContext.BinaryClassification.Trainers.LogisticRegression(
48-
new LogisticRegressionBinaryTrainer.Options { NumberOfThreads = 1 }));
47+
.Append(mlContext.BinaryClassification.Trainers.LbfgsLogisticRegression(
48+
new LbfgsLogisticRegressionBinaryTrainer.Options { NumberOfThreads = 1 }));
4949

5050
// Train the model.
5151
var model = pipeline.Fit(data);
@@ -74,10 +74,9 @@ public void ReconfigurablePrediction()
7474
public void ReconfigurablePredictionNoPipeline()
7575
{
7676
var mlContext = new MLContext(seed: 1);
77-
7877
var data = mlContext.Data.LoadFromEnumerable(TypeTestData.GenerateDataset());
79-
var pipeline = mlContext.BinaryClassification.Trainers.LogisticRegression(
80-
new Trainers.LogisticRegressionBinaryTrainer.Options { NumberOfThreads = 1 });
78+
var pipeline = mlContext.BinaryClassification.Trainers.LbfgsLogisticRegression(
79+
new Trainers.LbfgsLogisticRegressionBinaryTrainer.Options { NumberOfThreads = 1 });
8180
var model = pipeline.Fit(data);
8281
var newModel = mlContext.BinaryClassification.ChangeModelThreshold(model, -2.0f);
8382
var rnd = new Random(1);

0 commit comments

Comments
 (0)