|
28 | 28 | // </copyright>
|
29 | 29 |
|
30 | 30 | using System;
|
| 31 | +using System.Diagnostics; |
31 | 32 | using System.Linq;
|
32 | 33 |
|
33 | 34 | using MathNet.Numerics.LinearRegression;
|
@@ -98,6 +99,55 @@ public void FitsToBestLineThroughOrigin()
|
98 | 99 | Assert.AreEqual(-0.467791, respSeq, 1e-4);
|
99 | 100 | }
|
100 | 101 |
|
| 102 | + [Test] |
| 103 | + public void FitsToBestLineThroughOriginSpanInterface() |
| 104 | + { |
| 105 | + // Mathematica: Fit[{{1,4.986},{2,2.347},{3,2.061},{4,-2.995},{5,-2.352},{6,-5.782}}, {x}, x] |
| 106 | + // -> -0.467791 x |
| 107 | + |
| 108 | + var x = Enumerable.Range(1, 6).Select(Convert.ToDouble).ToArray().AsSpan(); |
| 109 | + var y = new Span<double> (new[] { 4.986, 2.347, 2.061, -2.995, -2.352, -5.782 }); |
| 110 | + |
| 111 | + var resp = SimpleRegression.FitThroughOrigin(x, y); |
| 112 | + Assert.AreEqual(-0.467791, resp, 1e-4); |
| 113 | + } |
| 114 | + |
| 115 | + [TestCase(10000, 100)] |
| 116 | + public void FitsToBestLineThroughOriginBenchmarkSpanVsArray(int dataLength, int windowLength) |
| 117 | + { |
| 118 | + var xarray = new double[dataLength]; |
| 119 | + var yarray = new double[dataLength]; |
| 120 | + |
| 121 | + var window = windowLength; |
| 122 | + |
| 123 | + var resultArray = new double[dataLength]; |
| 124 | + var resultSpan = new double[dataLength]; |
| 125 | + |
| 126 | + Stopwatch stopwatchArraySlicing = Stopwatch.StartNew(); |
| 127 | + // Use Array Slicing |
| 128 | + for (int i = 0; i < xarray.Length - window; ++i) |
| 129 | + { |
| 130 | + var xx = xarray.Skip(i).Take(window).ToArray(); |
| 131 | + var yy = yarray.Skip(i).Take(window).ToArray(); |
| 132 | + resultArray[i] = SimpleRegression.FitThroughOrigin(xx, yy); |
| 133 | + } |
| 134 | + stopwatchArraySlicing.Stop(); |
| 135 | + |
| 136 | + Stopwatch stopwatchSpanSlicing = Stopwatch.StartNew(); |
| 137 | + // Use Span Slicing |
| 138 | + for (int i = 0; i < xarray.Length - window; ++i) |
| 139 | + { |
| 140 | + var xx = xarray.AsSpan(i, window); |
| 141 | + var yy = yarray.AsSpan(i, window); |
| 142 | + resultSpan[i] = SimpleRegression.FitThroughOrigin(xx, yy); |
| 143 | + } |
| 144 | + stopwatchSpanSlicing.Stop(); |
| 145 | + |
| 146 | + Assert.AreEqual(resultArray, resultSpan); |
| 147 | + |
| 148 | + Console.WriteLine($"Array Slicing Timing: {stopwatchArraySlicing.Elapsed.TotalMilliseconds} ms, Span Slicing Timing: {stopwatchSpanSlicing.Elapsed.TotalMilliseconds} ms"); |
| 149 | + } |
| 150 | + |
101 | 151 | [Test]
|
102 | 152 | public void FitsToLineSameAsExcelTrendLine()
|
103 | 153 | {
|
|
0 commit comments