Skip to content

Commit d03e55f

Browse files
authored
Print failed iterations in CLI (dotnet#275)
1 parent dba3828 commit d03e55f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/mlnet/Utilities/ConsolePrinter.cs

+3-3
Original file line numberDiff line numberDiff line change
@@ -14,17 +14,17 @@ internal class ConsolePrinter
1414

1515
internal static void PrintBinaryClassificationMetrics(int iteration, string trainerName, BinaryClassificationMetrics metrics)
1616
{
17-
logger.Log(LogLevel.Info, $"{iteration,-4} {trainerName,-35} {metrics.Accuracy,9:F4} {metrics.Auc,8:F4} {metrics.Auprc,8:F4} {metrics.F1Score,9:F4}");
17+
logger.Log(LogLevel.Info, $"{iteration,-4} {trainerName,-35} {metrics?.Accuracy ?? double.NaN,9:F4} {metrics?.Auc ?? double.NaN,8:F4} {metrics?.Auprc ?? double.NaN,8:F4} {metrics?.F1Score ?? double.NaN,9:F4}");
1818
}
1919

2020
internal static void PrintMulticlassClassificationMetrics(int iteration, string trainerName, MultiClassClassifierMetrics metrics)
2121
{
22-
logger.Log(LogLevel.Info, $"{iteration,-4} {trainerName,-35} {metrics.AccuracyMicro,14:F4} {metrics.AccuracyMacro,14:F4}");
22+
logger.Log(LogLevel.Info, $"{iteration,-4} {trainerName,-35} {metrics?.AccuracyMicro ?? double.NaN,14:F4} {metrics?.AccuracyMacro ?? double.NaN,14:F4}");
2323
}
2424

2525
internal static void PrintRegressionMetrics(int iteration, string trainerName, RegressionMetrics metrics)
2626
{
27-
logger.Log(LogLevel.Info, $"{iteration,-4} {trainerName,-35} {metrics.RSquared,9:F4} {metrics.LossFn,12:F2} {metrics.L1,15:F2} {metrics.L2,15:F2} {metrics.Rms,12:F2}");
27+
logger.Log(LogLevel.Info, $"{iteration,-4} {trainerName,-35} {metrics?.RSquared ?? double.NaN,9:F4} {metrics?.LossFn ?? double.NaN,12:F2} {metrics?.L1 ?? double.NaN,15:F2} {metrics?.L2 ?? double.NaN,15:F2} {metrics?.Rms ?? double.NaN,12:F2}");
2828
}
2929

3030
internal static void PrintBinaryClassificationMetricsHeader()

0 commit comments

Comments
 (0)