summaryrefslogtreecommitdiff
path: root/RhSolutions.ML.Tests/ModelTests.cs
blob: 436f2e6b973147558cb2e55688c706388aa01a6d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
using RhSolutions.ML.Lib;

namespace RhSolutions.ML.Tests;

public class ModelTests
{
	[Test]
	public void EvaluateTest()
	{
		var testMetrics = RhSolutionsMLBuilder.EvaluateModel();
		Console.WriteLine($"*************************************************************************************************************");
		Console.WriteLine($"*       Metrics for Multi-class Classification model - Test Data     ");
		Console.WriteLine($"*------------------------------------------------------------------------------------------------------------");
		Console.WriteLine($"*       MicroAccuracy:    {testMetrics.MicroAccuracy:0.###}");
		Console.WriteLine($"*       MacroAccuracy:    {testMetrics.MacroAccuracy:0.###}");
		Console.WriteLine($"*       LogLoss:          {testMetrics.LogLoss:#.###}");
		Console.WriteLine($"*       LogLossReduction: {testMetrics.LogLossReduction:#.###}");
		Console.WriteLine($"*************************************************************************************************************");
		Assert.That(testMetrics.MicroAccuracy, Is.AtLeast(0.95));
		Assert.That(testMetrics.MacroAccuracy, Is.AtLeast(0.95));
		Assert.That(testMetrics.LogLoss, Is.AtMost(0.2));
		Assert.That(testMetrics.LogLossReduction, Is.AtLeast(0.80));
	}
}