aboutsummaryrefslogtreecommitdiffstats
path: root/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/distance/KSDistance.xtend
blob: 1fb21529f02e551d7b3cc976849f18bcb9536b01 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.distance

import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.app.Domain
import java.util.List
import org.apache.commons.math3.stat.inference.KolmogorovSmirnovTest
import ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.io.RepMetricsReader

class KSDistance {
	var static ksTester = new KolmogorovSmirnovTest();
	var double[] mpcSamples;
	var double[] naSamples;
	var double[] outDegreeSamples;
	
	new(Domain d){
		var metrics = RepMetricsReader.read(d);
		mpcSamples = metrics.mpcSamples;
		naSamples = metrics.naSamples.stream.mapToDouble([it]).toArray();
		outDegreeSamples = metrics.outDegreeSamples.stream.mapToDouble([it]).toArray();
	}
	
	def double mpcDistance(List<Double> samples){
		// map list to array
		var arr = samples.stream.mapToDouble([it]).toArray();
		
		//if the size of array is smaller than 2, ks distance cannot be performed, simply return 1
		if(arr.size < 2) return 1;
		return ksTester.kolmogorovSmirnovStatistic(mpcSamples, arr);
	}
	
	def double naDistance(List<Double> samples){
		// map list to array
		var arr = samples.stream.mapToDouble([it]).toArray();
		
		//if the size of array is smaller than 2, ks distance cannot be performed, simply return 1
		if(arr.size < 2) return 1;
		return ksTester.kolmogorovSmirnovStatistic(naSamples as double[], arr);
	}
	
	def double outDegreeDistance(List<Double> samples){
		// map list to array
		var arr = samples.stream.mapToDouble([it]).toArray();
		
		//if the size of array is smaller than 2, ks distance cannot be performed, simply return 1
		if(arr.size < 2) return 1;
		return ksTester.kolmogorovSmirnovStatistic(outDegreeSamples, arr);
	}
}