Changeset 18068 for branches/3136_Structural_GP/HeuristicLab.Problems.DataAnalysis.Symbolic.Regression

Timestamp:

10/15/21 17:00:28 (3 years ago)

Author:

dpiringe

Message:

#3136

• modified the StructureTemplateView to enable colorful tree nodes of type SubFunctionTreeNode
• refactored SubFunctionTreeNode, SubFunction and StructureTemplate

File:

• branches/3136_Structural_GP/HeuristicLab.Problems.DataAnalysis.Symbolic.Regression/3.4/SingleObjective/StructuredSymbolicRegressionSingleObjectiveProblem.cs (modified) (9 diffs)

branches/3136_Structural_GP/HeuristicLab.Problems.DataAnalysis.Symbolic.Regression/3.4/SingleObjective/StructuredSymbolicRegressionSingleObjectiveProblem.cs

@@ -22 +22 @@
     private const string ProblemDataParameterName = "ProblemData";
     private const string StructureDefinitionParameterName = "Structure Definition";
-    private const string GrammarParameterName = "Grammar";
     private const string StructureTemplateParameterName = "Structure Template";
     #endregion
@@ -30 +29 @@
     public IFixedValueParameter<StringValue> StructureDefinitionParameter => (IFixedValueParameter<StringValue>)Parameters[StructureDefinitionParameterName];
     public IFixedValueParameter<StructureTemplate> StructureTemplateParameter => (IFixedValueParameter<StructureTemplate>)Parameters[StructureTemplateParameterName];
-    public IValueParameter<ISymbolicDataAnalysisGrammar> GrammarParameter => (IValueParameter<ISymbolicDataAnalysisGrammar>)Parameters[GrammarParameterName];
     #endregion
 
@@ -51 +49 @@
     }
 
-    public ISymbolicDataAnalysisGrammar Grammar {
-      get => GrammarParameter.Value;
-      set => GrammarParameter.Value = value;
-    }
-
     IParameter IDataAnalysisProblem.ProblemDataParameter => ProblemDataParameter;
     IDataAnalysisProblemData IDataAnalysisProblem.ProblemData => ProblemData;
@@ -69 +62 @@
     public StructuredSymbolicRegressionSingleObjectiveProblem() {
       var problemData = new ShapeConstrainedRegressionProblemData();
-      var grammar = new LinearScalingGrammar();
-      var varSym = (Variable)grammar.GetSymbol("Variable");
-      varSym.AllVariableNames = problemData.InputVariables.Select(x => x.Value);
-      varSym.VariableNames = problemData.InputVariables.Select(x => x.Value);
-      varSym.Enabled = true;
 
       var structureTemplate = new StructureTemplate();
@@ -80 +68 @@
       Parameters.Add(new ValueParameter<IRegressionProblemData>(ProblemDataParameterName, problemData));
       Parameters.Add(new FixedValueParameter<StructureTemplate>(StructureTemplateParameterName, structureTemplate));
-      Parameters.Add(new ValueParameter<ISymbolicDataAnalysisGrammar>(GrammarParameterName, grammar));
 
-      //structureTemplate.Template = "f(x)*f(y)+5";
     }
 
@@ -97 +83 @@
 
     private void OnTemplateChanged(object sender, EventArgs args) {
+      SetupStructureTemplate();
+    }
+
+    private void SetupStructureTemplate() {
       foreach (var e in Encoding.Encodings.ToArray())
         Encoding.Remove(e);
 
-      foreach (var sf in StructureTemplate.SubFunctions.Values) {
-        Encoding.Add(new SymbolicExpressionTreeEncoding(sf.Name, sf.Grammar, sf.MaximumSymbolicExpressionTreeLength, sf.MaximumSymbolicExpressionTreeDepth));
+      foreach (var f in StructureTemplate.SubFunctions.Values) {
+        SetupVariables(f);
+        if(!Encoding.Encodings.Any(x => x.Name == f.Name)) // to prevent the same encoding twice
+          Encoding.Add(new SymbolicExpressionTreeEncoding(f.Name, f.Grammar, f.MaximumSymbolicExpressionTreeLength, f.MaximumSymbolicExpressionTreeDepth));
       }
     }
@@ -122 +114 @@
       else
         results.Add(new Result("Best Tree", BuildTree(individuals[bestIdx])));
-
-      /*
-      if (results.TryGetValue("Tree", out IResult result)) {
-        var list = result.Value as ItemList<ISymbolicExpressionTree>;
-        list.Clear();
-        list.AddRange(individuals.Select(x => (BuildTree(x))));
-      } else 
-        results.Add(new Result("Tree", new ItemList<ISymbolicExpressionTree>(individuals.Select(x => (BuildTree(x))))));
-      */
     }
 
@@ -152 +135 @@
         if (n.Symbol is SubFunctionSymbol) {
           var subFunctionTreeNode = n as SubFunctionTreeNode;
-          var subFunctionTree = individual.SymbolicExpressionTree(subFunctionTreeNode.SubFunction.Name);
+          var subFunctionTree = individual.SymbolicExpressionTree(subFunctionTreeNode.Name);
           var parent = n.Parent;
 
@@ -167 +150 @@
     }
 
+    private void SetupVariables(SubFunction subFunction) {
+      var varSym = (Variable)subFunction.Grammar.GetSymbol("Variable");
+      if (varSym == null) {
+        varSym = new Variable();
+        subFunction.Grammar.AddSymbol(varSym);
+      }
+
+      var allVariables = ProblemData.InputVariables.Select(x => x.Value);
+      var allInputs = allVariables.Where(x => x != ProblemData.TargetVariable);
+
+      // set all variables
+      varSym.AllVariableNames = allVariables;
+
+      // set all allowed variables
+      if (subFunction.Arguments.Contains("_")) {
+        varSym.VariableNames = allInputs;
+      } else {
+        var vars = new List<string>();
+        var exceptions = new List<Exception>();
+        foreach (var arg in subFunction.Arguments) {
+          if (allInputs.Contains(arg))
+            vars.Add(arg);
+          else
+            exceptions.Add(new ArgumentException($"The argument '{arg}' for sub-function '{subFunction.Name}' is not a valid variable."));
+        }
+        if (exceptions.Any())
+          throw new AggregateException(exceptions);
+        varSym.VariableNames = vars;
+      }
+
+      varSym.Enabled = true;
+    }
+
     public void Load(RegressionProblemData data) {
       ProblemData = data;
+      SetupStructureTemplate();
     }
   }