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