Changeset 2034 for trunk/sources/HeuristicLab.Modeling

Timestamp:

06/09/09 14:34:56 (16 years ago)

Author:

gkronber

Message:

Implemented a first version of an operator to calculate variable impacts of models (generated by GP or SVM). #644 (Variable impact of CEDMA models should be calculated and stored in the result DB)

Location:

trunk/sources/HeuristicLab.Modeling/3.2

Files:

• HeuristicLab.Modeling-3.2.csproj (modified) (1 diff)
• IModel.cs (modified) (1 diff)
• Model.cs (modified) (2 diffs)
• VariableImpactCalculator.cs (added)

trunk/sources/HeuristicLab.Modeling/3.2/HeuristicLab.Modeling-3.2.csproj

@@ -83 +83 @@
   <ItemGroup>
     <Compile Include="ClassificationProblemInjector.cs" />
+    <Compile Include="VariableImpactCalculator.cs" />
     <Compile Include="Model.cs" />
     <Compile Include="IModel.cs" />

trunk/sources/HeuristicLab.Modeling/3.2/IModel.cs

@@ -45 +45 @@
     double ValidationVarianceAccountedFor { get; }
     double TestVarianceAccountedFor { get; }
+    double GetVariableImpact(string variableName);
 
     IItem Data { get; }

trunk/sources/HeuristicLab.Modeling/3.2/Model.cs

@@ -120 +120 @@
     }
 
+    public double GetVariableImpact(string variableName) {
+      if (variableImpacts.ContainsKey(variableName)) return variableImpacts[variableName];
+      else return 0.0;
+    }
+
     private IItem data;
     public IItem Data {
@@ -127 +132 @@
 
     #endregion
+
+    private Dictionary<string, double> variableImpacts = new Dictionary<string, double>();
+    public void SetVariableImpact(string variableName, double impact) {
+      variableImpacts[variableName] = impact;
+    }
+
+    public void SetVariableImpact(int variableIndex, double impact) {
+      variableImpacts[dataset.GetVariableName(variableIndex)] = impact;
+    }
   }
 }