diff --git a/src/mlnet/Templates/Console/ModelBuilder.tt b/src/mlnet/Templates/Console/ModelBuilder.tt index f2f0240369..cf0c7346fc 100644 --- a/src/mlnet/Templates/Console/ModelBuilder.tt +++ b/src/mlnet/Templates/Console/ModelBuilder.tt @@ -188,18 +188,18 @@ else{#> { var L1 = crossValidationResults.Select(r => r.Metrics.MeanAbsoluteError); var L2 = crossValidationResults.Select(r => r.Metrics.MeanSquaredError); - var RMS = crossValidationResults.Select(r => r.Metrics.MeanAbsoluteError); + var RMS = crossValidationResults.Select(r => r.Metrics.RootMeanSquaredError); var lossFunction = crossValidationResults.Select(r => r.Metrics.LossFunction); var R2 = crossValidationResults.Select(r => r.Metrics.RSquared); Console.WriteLine($"*************************************************************************************************************"); Console.WriteLine($"* Metrics for Regression model "); Console.WriteLine($"*------------------------------------------------------------------------------------------------------------"); - Console.WriteLine($"* Average L1 Loss: {L1.Average():0.###} "); - Console.WriteLine($"* Average L2 Loss: {L2.Average():0.###} "); - Console.WriteLine($"* Average RMS: {RMS.Average():0.###} "); + Console.WriteLine($"* Average L1 Loss: {L1.Average():0.###} "); + Console.WriteLine($"* Average L2 Loss: {L2.Average():0.###} "); + Console.WriteLine($"* Average RMS: {RMS.Average():0.###} "); Console.WriteLine($"* Average Loss Function: {lossFunction.Average():0.###} "); - Console.WriteLine($"* Average R-squared: {R2.Average():0.###} "); + Console.WriteLine($"* Average R-squared: {R2.Average():0.###} "); Console.WriteLine($"*************************************************************************************************************"); } <# } if("BinaryClassification".Equals(TaskType)){ #> diff --git a/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ConsoleAppModelBuilderCSFileContentRegressionTest.approved.txt b/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ConsoleAppModelBuilderCSFileContentRegressionTest.approved.txt index 8211bd7e61..7f71a0b7d5 100644 --- a/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ConsoleAppModelBuilderCSFileContentRegressionTest.approved.txt +++ b/test/mlnet.Tests/ApprovalTests/ConsoleCodeGeneratorTests.ConsoleAppModelBuilderCSFileContentRegressionTest.approved.txt @@ -119,18 +119,18 @@ namespace TestNamespace.ConsoleApp { var L1 = crossValidationResults.Select(r => r.Metrics.MeanAbsoluteError); var L2 = crossValidationResults.Select(r => r.Metrics.MeanSquaredError); - var RMS = crossValidationResults.Select(r => r.Metrics.MeanAbsoluteError); + var RMS = crossValidationResults.Select(r => r.Metrics.RootMeanSquaredError); var lossFunction = crossValidationResults.Select(r => r.Metrics.LossFunction); var R2 = crossValidationResults.Select(r => r.Metrics.RSquared); Console.WriteLine($"*************************************************************************************************************"); Console.WriteLine($"* Metrics for Regression model "); Console.WriteLine($"*------------------------------------------------------------------------------------------------------------"); - Console.WriteLine($"* Average L1 Loss: {L1.Average():0.###} "); - Console.WriteLine($"* Average L2 Loss: {L2.Average():0.###} "); - Console.WriteLine($"* Average RMS: {RMS.Average():0.###} "); + Console.WriteLine($"* Average L1 Loss: {L1.Average():0.###} "); + Console.WriteLine($"* Average L2 Loss: {L2.Average():0.###} "); + Console.WriteLine($"* Average RMS: {RMS.Average():0.###} "); Console.WriteLine($"* Average Loss Function: {lossFunction.Average():0.###} "); - Console.WriteLine($"* Average R-squared: {R2.Average():0.###} "); + Console.WriteLine($"* Average R-squared: {R2.Average():0.###} "); Console.WriteLine($"*************************************************************************************************************"); } }