1 files changed, 4 insertions, 11 deletions

diff --git a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/distance/CostDistance.xtend b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/distance/CostDistance.xtend
index ee856201..33d10fa3 100644
--- a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/distance/CostDistance.xtend
+++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/distance/CostDistance.xtend
@@ -1,28 +1,21 @@
 package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.distance
 
-import org.apache.commons.math3.stat.regression.SimpleRegression
 import org.eclipse.xtend.lib.annotations.Accessors
 
 class CostDistance {
-        
-        var SimpleRegression regression;
-        
-        new(){
-                regression = new SimpleRegression(true);
-        }
-        
+
 }
 
 class StateData{
         @Accessors(PUBLIC_GETTER)
-        var double numOfNodeFeature;
+        var double[] features;
         @Accessors(PUBLIC_GETTER)
         var double value;
         @Accessors(PUBLIC_GETTER)
         var Object lastState;
         
-        new(int numOfNode, double value, Object lastState){
-                this.numOfNodeFeature = 1.0 / numOfNode;
+        new(double[] features, double value, Object lastState){
+                this.features = features;
                 this.value = value
                 this.lastState = lastState;
         }