Changeset 15297 for branches/dataset-ids-2695/HeuristicLab.Problems.DataAnalysis/3.4/Implementation

Timestamp:

07/29/17 16:03:05 (7 years ago)

Author:

gkronber

Message:

#2695: added a parameter to DataAnalysisProblemData to select an id column and extended RegressionLinechartViewBase to show the id on the xAxis

File:

• branches/dataset-ids-2695/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/DataAnalysisProblemData.cs (modified) (7 diffs)

branches/dataset-ids-2695/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/DataAnalysisProblemData.cs

@@ -21 +21 @@
 
 using System;
+using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
@@ -39 +40 @@
     protected const string TestPartitionParameterName = "TestPartition";
     protected const string TransformationsParameterName = "Transformations";
-
+    protected const string IdVariableParameterName = "Id Variable";
+    protected const string EmptyIdVariableName = "-";
     #region parameter properites
     //mkommend: inserted parameter caching due to performance reasons
@@ -58 +60 @@
     }
 
+    private IConstrainedValueParameter<StringValue> idVariableParameter;
+    public IConstrainedValueParameter<StringValue> IdVariableParameter {
+      get {
+        if (idVariableParameter == null) idVariableParameter = (IConstrainedValueParameter<StringValue>)Parameters[IdVariableParameterName];
+        return idVariableParameter;
+      }
+    }
+
     private IFixedValueParameter<IntRange> trainingPartitionParameter;
     public IFixedValueParameter<IntRange> TrainingPartitionParameter {
@@ -92 +102 @@
     public IEnumerable<string> AllowedInputVariables {
       get { return InputVariables.CheckedItems.Select(x => x.Value.Value); }
+    }
+    public string IdVariable {
+      get { return IdVariableParameter.Value.Value; }
+      // set only possible via parameter
+    }
+    public IEnumerable AllIds {
+      get {
+        var idVar = IdVariable;
+        if (idVar == EmptyIdVariableName) return Enumerable.Range(0, Dataset.Rows);
+        else if (Dataset.VariableHasType<double>(idVar)) return Dataset.GetDoubleValues(idVar);
+        else if (Dataset.VariableHasType<DateTime>(idVar)) return Dataset.GetDateTimeValues(idVar);
+        else if (Dataset.VariableHasType<string>(idVar)) return Dataset.GetStringValues(idVar);
+        else throw new NotSupportedException("The data type of the id variable is not supported");
+      }
+    }
+    public IEnumerable TrainingIds {
+      get {
+        var idVar = IdVariable;
+        if (idVar == EmptyIdVariableName) return TrainingIndices;
+        else if (Dataset.VariableHasType<double>(idVar)) return Dataset.GetDoubleValues(idVar, TrainingIndices);
+        else if (Dataset.VariableHasType<DateTime>(idVar)) return Dataset.GetDateTimeValues(idVar, TrainingIndices);
+        else if (Dataset.VariableHasType<string>(idVar)) return Dataset.GetStringValues(idVar, TrainingIndices);
+        else throw new NotSupportedException("The data type of the id variable is not supported");
+      }
+    }
+    public IEnumerable TestIds {
+      get {
+        var idVar = IdVariable;
+        if (idVar == EmptyIdVariableName) return TestIndices;
+        else if (Dataset.VariableHasType<double>(idVar)) return Dataset.GetDoubleValues(idVar, TestIndices);
+        else if (Dataset.VariableHasType<DateTime>(idVar)) return Dataset.GetDateTimeValues(idVar, TestIndices);
+        else if (Dataset.VariableHasType<string>(idVar)) return Dataset.GetStringValues(idVar, TestIndices);
+        else throw new NotSupportedException("The data type of the id variable is not supported");
+      }
     }
 
@@ -152 +196 @@
         TransformationsParameter.Hidden = true;
       }
+      if (!Parameters.ContainsKey(IdVariableParameterName)) {
+        var allIdVars = new ItemSet<StringValue>(GetIdVariableNames());
+        var idVar = allIdVars.First();
+        Parameters.Add(new ConstrainedValueParameter<StringValue>(IdVariableParameterName, "", allIdVars, idVar));
+        TransformationsParameter.Hidden = true;
+      }
       RegisterEventHandlers();
+    }
+
+    private IEnumerable<StringValue> GetIdVariableNames() {
+      // all variables with number of different values == number of rows
+      // + a separate entry for row id
+      // select separate entry for row id as default
+      // use event to update when the dataset is changed
+      var nRows = Dataset.Rows;
+      var potentialDoubleIdVars = Dataset.DoubleVariables
+        .Where(vn => Dataset.GetReadOnlyDoubleValues(vn).Distinct().Count() == nRows).ToArray();
+      var potentialStringIdVars = Dataset.StringVariables
+        .Where(vn => Dataset.GetReadOnlyStringValues(vn).Distinct().Count() == nRows).ToArray();
+      var potentialDateTimeIdVars = Dataset.DateTimeVariables
+        .Where(vn => Dataset.GetReadOnlyDateTimeValues(vn).Distinct().Count() == nRows).ToArray();
+      return
+        new string[] { EmptyIdVariableName }
+        .Concat(potentialStringIdVars)
+        .Concat(potentialDateTimeIdVars)
+        .Concat(potentialDoubleIdVars)
+        .Select(s => new StringValue(s));
     }
 
@@ -179 +249 @@
       Parameters.Add(new FixedValueParameter<IntRange>(TestPartitionParameterName, "", new IntRange(testPartitionStart, testPartitionEnd)));
       Parameters.Add(new FixedValueParameter<ReadOnlyItemList<ITransformation>>(TransformationsParameterName, "", transformationsList.AsReadOnly()));
-
       TransformationsParameter.Hidden = true;
+      var allIdVars = new ItemSet<StringValue>(GetIdVariableNames());
+      Parameters.Add(new ConstrainedValueParameter<StringValue>(IdVariableParameterName, "The variable to use as an ID column", allIdVars, allIdVars.First()));
 
       ((ValueParameter<Dataset>)DatasetParameter).ReactOnValueToStringChangedAndValueItemImageChanged = false;
@@ -199 +270 @@
 
     private void Parameter_ValueChanged(object sender, EventArgs e) {
+      // update list of possible id variables and set currently set id variable if possible
+      var curId = IdVariable;
+      var newIdVars = GetIdVariableNames().ToArray();
+      var matchingIdVar = newIdVars.FirstOrDefault(var => var.Value == curId);
+      if (matchingIdVar == null) matchingIdVar = newIdVars.First();
+      IdVariableParameter.ValidValues.Clear();
+      var validValues = IdVariableParameter.ValidValues;
+      foreach(var entry in newIdVars) validValues.Add(entry);
+      IdVariableParameter.Value = matchingIdVar;
+
       OnChanged();
     }