Skip to content

Commit cc7bb86

Browse files
authored
AdvancedExperimentSettings sample nits (dotnet#265)
1 parent 3326539 commit cc7bb86

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

src/Samples/AdvancedExperimentSettings.cs

+5-1
Original file line numberDiff line numberDiff line change
@@ -38,11 +38,15 @@ public static void Run()
3838
experimentSettings.ProgressHandler = new ProgressHandler();
3939

4040
// STEP 3: Using a different optimizing metric instead of RSquared and use only LightGbm
41-
Console.WriteLine($"Starting an experiment with MeanSquaredError optimizing metric and using LightGbm trainer only");
4241
experimentSettings.OptimizingMetric = RegressionMetric.MeanSquaredError;
4342
experimentSettings.Trainers.Clear();
4443
experimentSettings.Trainers.Add(RegressionTrainer.LightGbm);
4544

45+
// STEP 4: Start AutoML experiment
46+
Console.WriteLine($"Starting an experiment with MeanSquaredError optimizing metric and using LightGbm trainer only\r\n");
47+
RegressionExperiment autoExperiment = mlContext.Auto().CreateRegressionExperiment(experimentSettings);
48+
autoExperiment.Execute(trainDataView, LabelColumn);
49+
4650
Console.WriteLine("Press any key to continue...");
4751
Console.ReadKey();
4852
}

src/Samples/ObserveProgress.cs

+7-1
Original file line numberDiff line numberDiff line change
@@ -47,13 +47,19 @@ public static void Run()
4747
class ProgressHandler : IProgress<RunResult<RegressionMetrics>>
4848
{
4949
int iterationIndex;
50+
private bool _initialized = false;
51+
5052
public ProgressHandler()
5153
{
52-
ConsolePrinter.PrintRegressionMetricsHeader();
5354
}
5455

5556
public void Report(RunResult<RegressionMetrics> iterationResult)
5657
{
58+
if (!_initialized)
59+
{
60+
ConsolePrinter.PrintRegressionMetricsHeader();
61+
_initialized = true;
62+
}
5763
iterationIndex++;
5864
ConsolePrinter.PrintRegressionMetrics(iterationIndex, iterationResult.TrainerName, iterationResult.ValidationMetrics);
5965
}

0 commit comments

Comments
 (0)