forked from tsai/mathnet-numerics
Browse Source
Removed AlmostZero from incomplete Gamma function. Ported Pearson correlation computation. Added more histogram unit tests. Fixed a bug in the histogram code. Signed-off-by: jvangael <jurgen.vangael@gmail.com> Signed-off-by: jvangael <jurgen.vangael@gmail.com>la-knuth
9 changed files with 240 additions and 8 deletions
@ -0,0 +1,43 @@ |
|||||
|
// <copyright file="Histogram.fs" company="Math.NET"> |
||||
|
// Math.NET Numerics, part of the Math.NET Project |
||||
|
// http://mathnet.opensourcedotnet.info |
||||
|
// |
||||
|
// Copyright (c) 2009 Math.NET |
||||
|
// |
||||
|
// Permission is hereby granted, free of charge, to any person |
||||
|
// obtaining a copy of this software and associated documentation |
||||
|
// files (the "Software"), to deal in the Software without |
||||
|
// restriction, including without limitation the rights to use, |
||||
|
// copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
|
// copies of the Software, and to permit persons to whom the |
||||
|
// Software is furnished to do so, subject to the following |
||||
|
// conditions: |
||||
|
// |
||||
|
// The above copyright notice and this permission notice shall be |
||||
|
// included in all copies or substantial portions of the Software. |
||||
|
// |
||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
||||
|
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES |
||||
|
// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND |
||||
|
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT |
||||
|
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, |
||||
|
// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
||||
|
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR |
||||
|
// OTHER DEALINGS IN THE SOFTWARE. |
||||
|
// </copyright> |
||||
|
|
||||
|
open MathNet.Numerics.Statistics |
||||
|
|
||||
|
/// The number of buckets to use in our histogram. |
||||
|
let B = 9 |
||||
|
|
||||
|
/// Create a small dataset. |
||||
|
let data = [| 0.5; 1.5; 2.5; 3.5; 4.5; 5.5; 6.5; 7.5; 8.5; 9.5 |] |
||||
|
|
||||
|
/// A histogram with 9 buckets for this dataset. |
||||
|
let hist = new Histogram(data, B) |
||||
|
|
||||
|
// Print some histogram information. |
||||
|
printfn "Histogram.ToString(): %O" hist |
||||
|
for i in 0 .. B-1 do |
||||
|
printfn "Bucket %d contains %f datapoints." i hist.[i].Count |
||||
@ -0,0 +1,76 @@ |
|||||
|
// <copyright file="Correlation.cs" company="Math.NET">
|
||||
|
// Math.NET Numerics, part of the Math.NET Project
|
||||
|
// http://mathnet.opensourcedotnet.info
|
||||
|
//
|
||||
|
// Copyright (c) 2009 Math.NET
|
||||
|
//
|
||||
|
// Permission is hereby granted, free of charge, to any person
|
||||
|
// obtaining a copy of this software and associated documentation
|
||||
|
// files (the "Software"), to deal in the Software without
|
||||
|
// restriction, including without limitation the rights to use,
|
||||
|
// copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
|
// copies of the Software, and to permit persons to whom the
|
||||
|
// Software is furnished to do so, subject to the following
|
||||
|
// conditions:
|
||||
|
//
|
||||
|
// The above copyright notice and this permission notice shall be
|
||||
|
// included in all copies or substantial portions of the Software.
|
||||
|
//
|
||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||||
|
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
||||
|
// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||||
|
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
||||
|
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
||||
|
// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||
|
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
||||
|
// OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
// </copyright>
|
||||
|
|
||||
|
namespace MathNet.Numerics.Statistics |
||||
|
{ |
||||
|
using System; |
||||
|
using System.Collections.Generic; |
||||
|
using Properties; |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// A class with correlation measures between two datasets.
|
||||
|
/// </summary>
|
||||
|
public static class Correlation |
||||
|
{ |
||||
|
/// <summary>
|
||||
|
/// Computes the Pearson product-moment correlation coefficient.
|
||||
|
/// </summary>
|
||||
|
/// <param name="dataA">Sample data A.</param>
|
||||
|
/// <param name="dataB">Sample data B.</param>
|
||||
|
/// <returns>The Pearson product-moment correlation coefficient.</returns>
|
||||
|
public static double Pearson(IEnumerable<double> dataA, IEnumerable<double> dataB) |
||||
|
{ |
||||
|
int n = 0; |
||||
|
double r = 0.0; |
||||
|
double meanA = dataA.Mean(); |
||||
|
double meanB = dataB.Mean(); |
||||
|
double sdevA = dataA.StandardDeviation(); |
||||
|
double sdevB = dataB.StandardDeviation(); |
||||
|
|
||||
|
IEnumerator<double> ieA = dataA.GetEnumerator(); |
||||
|
IEnumerator<double> ieB = dataB.GetEnumerator(); |
||||
|
|
||||
|
while (ieA.MoveNext()) |
||||
|
{ |
||||
|
if (ieB.MoveNext() == false) |
||||
|
{ |
||||
|
throw new ArgumentOutOfRangeException("Datasets dataA and dataB need to have the same length."); |
||||
|
} |
||||
|
|
||||
|
n++; |
||||
|
r += (ieA.Current - meanA) * (ieB.Current - meanB) / (sdevA * sdevB); |
||||
|
} |
||||
|
if (ieB.MoveNext() == true) |
||||
|
{ |
||||
|
throw new ArgumentOutOfRangeException("Datasets dataA and dataB need to have the same length."); |
||||
|
} |
||||
|
|
||||
|
return r / (n - 1); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,70 @@ |
|||||
|
// <copyright file="CorrelationTests.cs" company="Math.NET">
|
||||
|
// Math.NET Numerics, part of the Math.NET Project
|
||||
|
// http://mathnet.opensourcedotnet.info
|
||||
|
//
|
||||
|
// Copyright (c) 2009 Math.NET
|
||||
|
//
|
||||
|
// Permission is hereby granted, free of charge, to any person
|
||||
|
// obtaining a copy of this software and associated documentation
|
||||
|
// files (the "Software"), to deal in the Software without
|
||||
|
// restriction, including without limitation the rights to use,
|
||||
|
// copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
|
// copies of the Software, and to permit persons to whom the
|
||||
|
// Software is furnished to do so, subject to the following
|
||||
|
// conditions:
|
||||
|
//
|
||||
|
// The above copyright notice and this permission notice shall be
|
||||
|
// included in all copies or substantial portions of the Software.
|
||||
|
//
|
||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||||
|
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
||||
|
// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||||
|
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
||||
|
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
||||
|
// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||
|
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
||||
|
// OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
// </copyright>
|
||||
|
|
||||
|
namespace MathNet.Numerics.UnitTests.StatisticsTests |
||||
|
{ |
||||
|
using System; |
||||
|
using System.Linq; |
||||
|
using System.Collections.Generic; |
||||
|
using MbUnit.Framework; |
||||
|
using MathNet.Numerics.Statistics; |
||||
|
using MathNet.Numerics.LinearAlgebra.Double; |
||||
|
|
||||
|
[TestFixture] |
||||
|
public class CorrelationTests |
||||
|
{ |
||||
|
private readonly IDictionary<string, StatTestData> mData = new Dictionary<string, StatTestData>(); |
||||
|
|
||||
|
public CorrelationTests() |
||||
|
{ |
||||
|
StatTestData lottery = new StatTestData("./data/NIST/Lottery.dat"); |
||||
|
mData.Add("lottery", lottery); |
||||
|
StatTestData lew = new StatTestData("./data/NIST/Lew.dat"); |
||||
|
mData.Add("lew", lew); |
||||
|
} |
||||
|
|
||||
|
[Test] |
||||
|
public void PearsonCorrelationTest() |
||||
|
{ |
||||
|
var dataA = mData["lottery"].Data.Take(200); |
||||
|
var dataB = mData["lew"].Data.Take(200); |
||||
|
|
||||
|
double corr = Correlation.Pearson(dataA, dataB); |
||||
|
AssertHelpers.AlmostEqual(corr, -0.029470861580726, 13); |
||||
|
} |
||||
|
|
||||
|
[Test] |
||||
|
public void PearsonCorrelationTest_Fail() |
||||
|
{ |
||||
|
var dataA = mData["lottery"].Data; |
||||
|
var dataB = mData["lew"].Data; |
||||
|
|
||||
|
Assert.Throws<ArgumentOutOfRangeException>(() => Correlation.Pearson(dataA, dataB)); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue