Changeset 10383 for branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.3/ProblemDataCreator.cs

Timestamp:

01/22/14 16:53:31 (11 years ago)

Author:

pfleck

Message:

• Added ProblemDataCreator for instancing a new DataAnalysisProblemData with changed Dataset etc.
• Added export functionality to PreprocessingContext. (cloned Algorithm or Problem)
• Commented out code in StatisticsLogic which breaks the build. :(

File:

• branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.3/ProblemDataCreator.cs (moved) (moved from branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.3/PreprocessingCloner.cs) (1 diff)

branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.3/ProblemDataCreator.cs

@@ -20 +20 @@
 #endregion
 
+using System;
 using System.Linq;
 using HeuristicLab.Common;
 using HeuristicLab.Core;
 using HeuristicLab.Data;
-using HeuristicLab.Optimization;
 using HeuristicLab.Problems.DataAnalysis;
 
 namespace HeuristicLab.DataPreprocessing {
-  internal class PreprocessingCloner {
+  internal class ProblemDataCreator {
 
     private readonly IPreprocessingContext context;
 
-    public PreprocessingCloner(IPreprocessingContext context) {
+    public ProblemDataCreator(IPreprocessingContext context) {
       this.context = context;
     }
 
-    public IItem GenerateAlteredClone(IItem obj) {
-      IPreprocessingData data = context.Data;
+    public IDataAnalysisProblemData CreateProblemData() {
 
-      Dataset dataset = data.ExportToDataset();
-      var variables = new CheckedItemList<StringValue>(data.VariableNames.Select(s => new StringValue(s))).AsReadOnly();
+      IDataAnalysisProblemData problemData = CloneProblemDataWithDataset();
 
+      SetTrainingAndTestPartition(problemData);
+
+      if (problemData is RegressionProblemData) {
+        //SetRegressionData((RegressionProblemData)problemData);
+      } else if (problemData is ClassificationProblemData) {
+        //SetClassificationData((ClassificationProblemData)problemData);
+      } else if (problemData is ClusteringProblemData) {
+        throw new NotImplementedException();
+      }
+
+      return problemData;
+    }
+
+    private IDataAnalysisProblemData CloneProblemDataWithDataset() {
       var cloner = new Cloner();
 
-      cloner.RegisterClonedObject(context.DataAnalysisProblemData.Dataset, dataset);
-      cloner.RegisterClonedObject(context.DataAnalysisProblemData.TrainingPartition, data.TrainingPartition);
-      cloner.RegisterClonedObject(context.DataAnalysisProblemData.TestPartition, data.TestPartition);
-      cloner.RegisterClonedObject(context.DataAnalysisProblemData.InputVariables, variables);
-      if (context.Algorithm != null) {
-        cloner.RegisterClonedObject(context.Algorithm.Runs, new RunCollection { OptimizerName = context.Algorithm.Name });
+      var problemData = context.Problem.ProblemData;
+
+      Dataset dataset = context.Data.ExportToDataset();
+
+      cloner.RegisterClonedObject(problemData.Dataset, dataset);
+
+      return cloner.Clone(problemData);
+    }
+
+    private void SetTrainingAndTestPartition(IDataAnalysisProblemData problemData) {
+      var ppData = context.Data;
+
+      problemData.TrainingPartition.Start = ppData.TrainingPartition.Start;
+      problemData.TrainingPartition.End = ppData.TrainingPartition.End;
+      problemData.TestPartition.Start = ppData.TestPartition.Start;
+      problemData.TestPartition.End = ppData.TestPartition.End;
+    }
+
+    private void SetRegressionData(RegressionProblemData regressionProblemData) {
+      SetInputVariables(regressionProblemData, regressionProblemData.TargetVariable);
+      SetTargetVariable(regressionProblemData.TargetVariableParameter);
+    }
+
+    private void SetClassificationData(ClassificationProblemData classificationProblemData) {
+      SetInputVariables(classificationProblemData, classificationProblemData.TargetVariable);
+      SetTargetVariable(classificationProblemData.TargetVariableParameter);
+    }
+
+    private void SetInputVariables(DataAnalysisProblemData problemData, string targetVariable) {
+      //TODO: InputVariables Set is Readonly
+
+      problemData.InputVariables.Clear();
+
+      foreach (string variable in context.Data.VariableNames) {
+        problemData.InputVariables.Add(new StringValue(variable), variable != targetVariable);
+      }
+    }
+
+    private void SetTargetVariable(IConstrainedValueParameter<StringValue> targetVariableParameter) {
+      string oldTarget = targetVariableParameter.Value.Value;
+
+      var validValues = targetVariableParameter.ValidValues;
+      validValues.Clear();
+
+      foreach (string variable in context.Data.VariableNames.Where(x => context.Data.IsType<double>(x))) {
+        validValues.Add(new StringValue(variable));
       }
 
-      return cloner.Clone(obj);
+      targetVariableParameter.ActualValue = validValues.FirstOrDefault(v => v.Value == oldTarget);
     }
   }