source: trunk/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/Interpreter/SymbolicDataAnalysisExpressionTreeLinearInterpreter.cs @ 16356

Last change on this file since 16356 was 16356, checked in by gkronber, 9 months ago

#2915: merged all changes from branch to trunk

File size: 22.2 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2018 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
4 *
5 * This file is part of HeuristicLab.
6 *
7 * HeuristicLab is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation, either version 3 of the License, or
10 * (at your option) any later version.
11 *
12 * HeuristicLab is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
19 */
20#endregion
21
22using System;
23using System.Collections.Generic;
24using System.Linq;
25using HeuristicLab.Common;
26using HeuristicLab.Core;
27using HeuristicLab.Data;
28using HeuristicLab.Encodings.SymbolicExpressionTreeEncoding;
29using HeuristicLab.Parameters;
30using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
31
32namespace HeuristicLab.Problems.DataAnalysis.Symbolic {
33  [StorableClass]
34  [Item("SymbolicDataAnalysisExpressionTreeLinearInterpreter", "Fast linear (non-recursive) interpreter for symbolic expression trees. Does not support ADFs.")]
35  public sealed class SymbolicDataAnalysisExpressionTreeLinearInterpreter : ParameterizedNamedItem, ISymbolicDataAnalysisExpressionTreeInterpreter {
36    private const string CheckExpressionsWithIntervalArithmeticParameterName = "CheckExpressionsWithIntervalArithmetic";
37    private const string CheckExpressionsWithIntervalArithmeticParameterDescription = "Switch that determines if the interpreter checks the validity of expressions with interval arithmetic before evaluating the expression.";
38    private const string EvaluatedSolutionsParameterName = "EvaluatedSolutions";
39
40    private readonly SymbolicDataAnalysisExpressionTreeInterpreter interpreter;
41
42    public override bool CanChangeName {
43      get { return false; }
44    }
45
46    public override bool CanChangeDescription {
47      get { return false; }
48    }
49
50    #region parameter properties
51    public IFixedValueParameter<BoolValue> CheckExpressionsWithIntervalArithmeticParameter {
52      get { return (IFixedValueParameter<BoolValue>)Parameters[CheckExpressionsWithIntervalArithmeticParameterName]; }
53    }
54
55    public IFixedValueParameter<IntValue> EvaluatedSolutionsParameter {
56      get { return (IFixedValueParameter<IntValue>)Parameters[EvaluatedSolutionsParameterName]; }
57    }
58    #endregion
59
60    #region properties
61    public bool CheckExpressionsWithIntervalArithmetic {
62      get { return CheckExpressionsWithIntervalArithmeticParameter.Value.Value; }
63      set { CheckExpressionsWithIntervalArithmeticParameter.Value.Value = value; }
64    }
65    public int EvaluatedSolutions {
66      get { return EvaluatedSolutionsParameter.Value.Value; }
67      set { EvaluatedSolutionsParameter.Value.Value = value; }
68    }
69    #endregion
70
71    [StorableConstructor]
72    private SymbolicDataAnalysisExpressionTreeLinearInterpreter(bool deserializing)
73      : base(deserializing) {
74      interpreter = new SymbolicDataAnalysisExpressionTreeInterpreter();
75    }
76
77    private SymbolicDataAnalysisExpressionTreeLinearInterpreter(SymbolicDataAnalysisExpressionTreeLinearInterpreter original, Cloner cloner)
78      : base(original, cloner) {
79      interpreter = cloner.Clone(original.interpreter);
80    }
81
82    public override IDeepCloneable Clone(Cloner cloner) {
83      return new SymbolicDataAnalysisExpressionTreeLinearInterpreter(this, cloner);
84    }
85
86    public SymbolicDataAnalysisExpressionTreeLinearInterpreter()
87      : base("SymbolicDataAnalysisExpressionTreeLinearInterpreter", "Linear (non-recursive) interpreter for symbolic expression trees (does not support ADFs).") {
88      Parameters.Add(new FixedValueParameter<BoolValue>(CheckExpressionsWithIntervalArithmeticParameterName, CheckExpressionsWithIntervalArithmeticParameterDescription, new BoolValue(false)));
89      Parameters.Add(new FixedValueParameter<IntValue>(EvaluatedSolutionsParameterName, "A counter for the total number of solutions the interpreter has evaluated", new IntValue(0)));
90      interpreter = new SymbolicDataAnalysisExpressionTreeInterpreter();
91    }
92
93    public SymbolicDataAnalysisExpressionTreeLinearInterpreter(string name, string description)
94      : base(name, description) {
95      Parameters.Add(new FixedValueParameter<BoolValue>(CheckExpressionsWithIntervalArithmeticParameterName, CheckExpressionsWithIntervalArithmeticParameterDescription, new BoolValue(false)));
96      Parameters.Add(new FixedValueParameter<IntValue>(EvaluatedSolutionsParameterName, "A counter for the total number of solutions the interpreter has evaluated", new IntValue(0)));
97      interpreter = new SymbolicDataAnalysisExpressionTreeInterpreter();
98    }
99
100    [StorableHook(HookType.AfterDeserialization)]
101    private void AfterDeserialization() {
102      var evaluatedSolutions = new IntValue(0);
103      var checkExpressionsWithIntervalArithmetic = new BoolValue(false);
104      if (Parameters.ContainsKey(EvaluatedSolutionsParameterName)) {
105        var evaluatedSolutionsParameter = (IValueParameter<IntValue>)Parameters[EvaluatedSolutionsParameterName];
106        evaluatedSolutions = evaluatedSolutionsParameter.Value;
107        Parameters.Remove(EvaluatedSolutionsParameterName);
108      }
109      Parameters.Add(new FixedValueParameter<IntValue>(EvaluatedSolutionsParameterName, "A counter for the total number of solutions the interpreter has evaluated", evaluatedSolutions));
110      if (Parameters.ContainsKey(CheckExpressionsWithIntervalArithmeticParameterName)) {
111        var checkExpressionsWithIntervalArithmeticParameter = (IValueParameter<BoolValue>)Parameters[CheckExpressionsWithIntervalArithmeticParameterName];
112        Parameters.Remove(CheckExpressionsWithIntervalArithmeticParameterName);
113        checkExpressionsWithIntervalArithmetic = checkExpressionsWithIntervalArithmeticParameter.Value;
114      }
115      Parameters.Add(new FixedValueParameter<BoolValue>(CheckExpressionsWithIntervalArithmeticParameterName, CheckExpressionsWithIntervalArithmeticParameterDescription, checkExpressionsWithIntervalArithmetic));
116    }
117
118    #region IStatefulItem
119    public void InitializeState() {
120      EvaluatedSolutions = 0;
121    }
122
123    public void ClearState() { }
124    #endregion
125
126    private readonly object syncRoot = new object();
127    public IEnumerable<double> GetSymbolicExpressionTreeValues(ISymbolicExpressionTree tree, IDataset dataset, IEnumerable<int> rows) {
128      if (!rows.Any()) return Enumerable.Empty<double>();
129      if (CheckExpressionsWithIntervalArithmetic)
130        throw new NotSupportedException("Interval arithmetic is not yet supported in the symbolic data analysis interpreter.");
131
132      lock (syncRoot) {
133        EvaluatedSolutions++; // increment the evaluated solutions counter
134      }
135
136      var code = SymbolicExpressionTreeLinearCompiler.Compile(tree, OpCodes.MapSymbolToOpCode);
137      PrepareInstructions(code, dataset);
138      return rows.Select(row => Evaluate(dataset, row, code));
139    }
140
141    private double Evaluate(IDataset dataset, int row, LinearInstruction[] code) {
142      for (int i = code.Length - 1; i >= 0; --i) {
143        if (code[i].skip) continue;
144        #region opcode if
145        var instr = code[i];
146        if (instr.opCode == OpCodes.Variable) {
147          if (row < 0 || row >= dataset.Rows) instr.value = double.NaN;
148          else {
149            var variableTreeNode = (VariableTreeNode)instr.dynamicNode;
150            instr.value = ((IList<double>)instr.data)[row] * variableTreeNode.Weight;
151          }
152        } else if (instr.opCode == OpCodes.BinaryFactorVariable) {
153          if (row < 0 || row >= dataset.Rows) instr.value = double.NaN;
154          else {
155            var factorTreeNode = instr.dynamicNode as BinaryFactorVariableTreeNode;
156            instr.value = ((IList<string>)instr.data)[row] == factorTreeNode.VariableValue ? factorTreeNode.Weight : 0;
157          }
158        } else if (instr.opCode == OpCodes.FactorVariable) {
159          if (row < 0 || row >= dataset.Rows) instr.value = double.NaN;
160          else {
161            var factorTreeNode = instr.dynamicNode as FactorVariableTreeNode;
162            instr.value = factorTreeNode.GetValue(((IList<string>)instr.data)[row]);
163          }
164        } else if (instr.opCode == OpCodes.LagVariable) {
165          var laggedVariableTreeNode = (LaggedVariableTreeNode)instr.dynamicNode;
166          int actualRow = row + laggedVariableTreeNode.Lag;
167          if (actualRow < 0 || actualRow >= dataset.Rows)
168            instr.value = double.NaN;
169          else
170            instr.value = ((IList<double>)instr.data)[actualRow] * laggedVariableTreeNode.Weight;
171        } else if (instr.opCode == OpCodes.VariableCondition) {
172          if (row < 0 || row >= dataset.Rows) instr.value = double.NaN;
173          var variableConditionTreeNode = (VariableConditionTreeNode)instr.dynamicNode;
174          if (!variableConditionTreeNode.Symbol.IgnoreSlope) {
175            double variableValue = ((IList<double>)instr.data)[row];
176            double x = variableValue - variableConditionTreeNode.Threshold;
177            double p = 1 / (1 + Math.Exp(-variableConditionTreeNode.Slope * x));
178
179            double trueBranch = code[instr.childIndex].value;
180            double falseBranch = code[instr.childIndex + 1].value;
181
182            instr.value = trueBranch * p + falseBranch * (1 - p);
183          } else {
184            double variableValue = ((IList<double>)instr.data)[row];
185            if (variableValue <= variableConditionTreeNode.Threshold) {
186              instr.value = code[instr.childIndex].value;
187            } else {
188              instr.value = code[instr.childIndex + 1].value;
189            }
190          }
191        } else if (instr.opCode == OpCodes.Add) {
192          double s = code[instr.childIndex].value;
193          for (int j = 1; j != instr.nArguments; ++j) {
194            s += code[instr.childIndex + j].value;
195          }
196          instr.value = s;
197        } else if (instr.opCode == OpCodes.Sub) {
198          double s = code[instr.childIndex].value;
199          for (int j = 1; j != instr.nArguments; ++j) {
200            s -= code[instr.childIndex + j].value;
201          }
202          if (instr.nArguments == 1) s = -s;
203          instr.value = s;
204        } else if (instr.opCode == OpCodes.Mul) {
205          double p = code[instr.childIndex].value;
206          for (int j = 1; j != instr.nArguments; ++j) {
207            p *= code[instr.childIndex + j].value;
208          }
209          instr.value = p;
210        } else if (instr.opCode == OpCodes.Div) {
211          double p = code[instr.childIndex].value;
212          for (int j = 1; j != instr.nArguments; ++j) {
213            p /= code[instr.childIndex + j].value;
214          }
215          if (instr.nArguments == 1) p = 1.0 / p;
216          instr.value = p;
217        } else if (instr.opCode == OpCodes.AnalyticalQuotient) {
218          var x1 = code[instr.childIndex].value;
219          var x2 = code[instr.childIndex + 1].value;
220          instr.value = x1 / Math.Sqrt(1 + x2 * x2);
221        } else if (instr.opCode == OpCodes.Average) {
222          double s = code[instr.childIndex].value;
223          for (int j = 1; j != instr.nArguments; ++j) {
224            s += code[instr.childIndex + j].value;
225          }
226          instr.value = s / instr.nArguments;
227        } else if (instr.opCode == OpCodes.Absolute) {
228          instr.value = Math.Abs(code[instr.childIndex].value);
229        } else if (instr.opCode == OpCodes.Cos) {
230          instr.value = Math.Cos(code[instr.childIndex].value);
231        } else if (instr.opCode == OpCodes.Sin) {
232          instr.value = Math.Sin(code[instr.childIndex].value);
233        } else if (instr.opCode == OpCodes.Tan) {
234          instr.value = Math.Tan(code[instr.childIndex].value);
235        } else if (instr.opCode == OpCodes.Square) {
236          instr.value = Math.Pow(code[instr.childIndex].value, 2);
237        } else if (instr.opCode == OpCodes.Cube) {
238          instr.value = Math.Pow(code[instr.childIndex].value, 3);
239        } else if (instr.opCode == OpCodes.Power) {
240          double x = code[instr.childIndex].value;
241          double y = Math.Round(code[instr.childIndex + 1].value);
242          instr.value = Math.Pow(x, y);
243        } else if (instr.opCode == OpCodes.SquareRoot) {
244          instr.value = Math.Sqrt(code[instr.childIndex].value);
245        } else if (instr.opCode == OpCodes.CubeRoot) {
246          instr.value = Math.Pow(code[instr.childIndex].value, 1.0 / 3.0);
247        } else if (instr.opCode == OpCodes.Root) {
248          double x = code[instr.childIndex].value;
249          double y = Math.Round(code[instr.childIndex + 1].value);
250          instr.value = Math.Pow(x, 1 / y);
251        } else if (instr.opCode == OpCodes.Exp) {
252          instr.value = Math.Exp(code[instr.childIndex].value);
253        } else if (instr.opCode == OpCodes.Log) {
254          instr.value = Math.Log(code[instr.childIndex].value);
255        } else if (instr.opCode == OpCodes.Gamma) {
256          var x = code[instr.childIndex].value;
257          instr.value = double.IsNaN(x) ? double.NaN : alglib.gammafunction(x);
258        } else if (instr.opCode == OpCodes.Psi) {
259          var x = code[instr.childIndex].value;
260          if (double.IsNaN(x)) instr.value = double.NaN;
261          else if (x <= 0 && (Math.Floor(x) - x).IsAlmost(0)) instr.value = double.NaN;
262          else instr.value = alglib.psi(x);
263        } else if (instr.opCode == OpCodes.Dawson) {
264          var x = code[instr.childIndex].value;
265          instr.value = double.IsNaN(x) ? double.NaN : alglib.dawsonintegral(x);
266        } else if (instr.opCode == OpCodes.ExponentialIntegralEi) {
267          var x = code[instr.childIndex].value;
268          instr.value = double.IsNaN(x) ? double.NaN : alglib.exponentialintegralei(x);
269        } else if (instr.opCode == OpCodes.SineIntegral) {
270          double si, ci;
271          var x = code[instr.childIndex].value;
272          if (double.IsNaN(x)) instr.value = double.NaN;
273          else {
274            alglib.sinecosineintegrals(x, out si, out ci);
275            instr.value = si;
276          }
277        } else if (instr.opCode == OpCodes.CosineIntegral) {
278          double si, ci;
279          var x = code[instr.childIndex].value;
280          if (double.IsNaN(x)) instr.value = double.NaN;
281          else {
282            alglib.sinecosineintegrals(x, out si, out ci);
283            instr.value = ci;
284          }
285        } else if (instr.opCode == OpCodes.HyperbolicSineIntegral) {
286          double shi, chi;
287          var x = code[instr.childIndex].value;
288          if (double.IsNaN(x)) instr.value = double.NaN;
289          else {
290            alglib.hyperbolicsinecosineintegrals(x, out shi, out chi);
291            instr.value = shi;
292          }
293        } else if (instr.opCode == OpCodes.HyperbolicCosineIntegral) {
294          double shi, chi;
295          var x = code[instr.childIndex].value;
296          if (double.IsNaN(x)) instr.value = double.NaN;
297          else {
298            alglib.hyperbolicsinecosineintegrals(x, out shi, out chi);
299            instr.value = chi;
300          }
301        } else if (instr.opCode == OpCodes.FresnelCosineIntegral) {
302          double c = 0, s = 0;
303          var x = code[instr.childIndex].value;
304          if (double.IsNaN(x)) instr.value = double.NaN;
305          else {
306            alglib.fresnelintegral(x, ref c, ref s);
307            instr.value = c;
308          }
309        } else if (instr.opCode == OpCodes.FresnelSineIntegral) {
310          double c = 0, s = 0;
311          var x = code[instr.childIndex].value;
312          if (double.IsNaN(x)) instr.value = double.NaN;
313          else {
314            alglib.fresnelintegral(x, ref c, ref s);
315            instr.value = s;
316          }
317        } else if (instr.opCode == OpCodes.AiryA) {
318          double ai, aip, bi, bip;
319          var x = code[instr.childIndex].value;
320          if (double.IsNaN(x)) instr.value = double.NaN;
321          else {
322            alglib.airy(x, out ai, out aip, out bi, out bip);
323            instr.value = ai;
324          }
325        } else if (instr.opCode == OpCodes.AiryB) {
326          double ai, aip, bi, bip;
327          var x = code[instr.childIndex].value;
328          if (double.IsNaN(x)) instr.value = double.NaN;
329          else {
330            alglib.airy(x, out ai, out aip, out bi, out bip);
331            instr.value = bi;
332          }
333        } else if (instr.opCode == OpCodes.Norm) {
334          var x = code[instr.childIndex].value;
335          if (double.IsNaN(x)) instr.value = double.NaN;
336          else instr.value = alglib.normaldistribution(x);
337        } else if (instr.opCode == OpCodes.Erf) {
338          var x = code[instr.childIndex].value;
339          if (double.IsNaN(x)) instr.value = double.NaN;
340          else instr.value = alglib.errorfunction(x);
341        } else if (instr.opCode == OpCodes.Bessel) {
342          var x = code[instr.childIndex].value;
343          if (double.IsNaN(x)) instr.value = double.NaN;
344          else instr.value = alglib.besseli0(x);
345        } else if (instr.opCode == OpCodes.IfThenElse) {
346          double condition = code[instr.childIndex].value;
347          double result;
348          if (condition > 0.0) {
349            result = code[instr.childIndex + 1].value;
350          } else {
351            result = code[instr.childIndex + 2].value;
352          }
353          instr.value = result;
354        } else if (instr.opCode == OpCodes.AND) {
355          double result = code[instr.childIndex].value;
356          for (int j = 1; j < instr.nArguments; j++) {
357            if (result > 0.0) result = code[instr.childIndex + j].value;
358            else break;
359          }
360          instr.value = result > 0.0 ? 1.0 : -1.0;
361        } else if (instr.opCode == OpCodes.OR) {
362          double result = code[instr.childIndex].value;
363          for (int j = 1; j < instr.nArguments; j++) {
364            if (result <= 0.0) result = code[instr.childIndex + j].value;
365            else break;
366          }
367          instr.value = result > 0.0 ? 1.0 : -1.0;
368        } else if (instr.opCode == OpCodes.NOT) {
369          instr.value = code[instr.childIndex].value > 0.0 ? -1.0 : 1.0;
370        } else if (instr.opCode == OpCodes.XOR) {
371          int positiveSignals = 0;
372          for (int j = 0; j < instr.nArguments; j++) {
373            if (code[instr.childIndex + j].value > 0.0) positiveSignals++;
374          }
375          instr.value = positiveSignals % 2 != 0 ? 1.0 : -1.0;
376        } else if (instr.opCode == OpCodes.GT) {
377          double x = code[instr.childIndex].value;
378          double y = code[instr.childIndex + 1].value;
379          instr.value = x > y ? 1.0 : -1.0;
380        } else if (instr.opCode == OpCodes.LT) {
381          double x = code[instr.childIndex].value;
382          double y = code[instr.childIndex + 1].value;
383          instr.value = x < y ? 1.0 : -1.0;
384        } else if (instr.opCode == OpCodes.TimeLag || instr.opCode == OpCodes.Derivative || instr.opCode == OpCodes.Integral) {
385          var state = (InterpreterState)instr.data;
386          state.Reset();
387          instr.value = interpreter.Evaluate(dataset, ref row, state);
388        } else {
389          var errorText = string.Format("The {0} symbol is not supported by the linear interpreter. To support this symbol, please use the SymbolicDataAnalysisExpressionTreeInterpreter.", instr.dynamicNode.Symbol.Name);
390          throw new NotSupportedException(errorText);
391        }
392        #endregion
393      }
394      return code[0].value;
395    }
396
397    private static LinearInstruction[] GetPrefixSequence(LinearInstruction[] code, int startIndex) {
398      var s = new Stack<int>();
399      var list = new List<LinearInstruction>();
400      s.Push(startIndex);
401      while (s.Any()) {
402        int i = s.Pop();
403        var instr = code[i];
404        // push instructions in reverse execution order
405        for (int j = instr.nArguments - 1; j >= 0; j--) s.Push(instr.childIndex + j);
406        list.Add(instr);
407      }
408      return list.ToArray();
409    }
410
411    public static void PrepareInstructions(LinearInstruction[] code, IDataset dataset) {
412      for (int i = 0; i != code.Length; ++i) {
413        var instr = code[i];
414        #region opcode switch
415        switch (instr.opCode) {
416          case OpCodes.Constant: {
417              var constTreeNode = (ConstantTreeNode)instr.dynamicNode;
418              instr.value = constTreeNode.Value;
419              instr.skip = true; // the value is already set so this instruction should be skipped in the evaluation phase
420            }
421            break;
422          case OpCodes.Variable: {
423              var variableTreeNode = (VariableTreeNode)instr.dynamicNode;
424              instr.data = dataset.GetReadOnlyDoubleValues(variableTreeNode.VariableName);
425            }
426            break;
427          case OpCodes.BinaryFactorVariable: {
428              var factorVariableTreeNode = instr.dynamicNode as BinaryFactorVariableTreeNode;
429              instr.data = dataset.GetReadOnlyStringValues(factorVariableTreeNode.VariableName);
430            }
431            break;
432          case OpCodes.FactorVariable: {
433              var factorVariableTreeNode = instr.dynamicNode as FactorVariableTreeNode;
434              instr.data = dataset.GetReadOnlyStringValues(factorVariableTreeNode.VariableName);
435            }
436            break;
437          case OpCodes.LagVariable: {
438              var laggedVariableTreeNode = (LaggedVariableTreeNode)instr.dynamicNode;
439              instr.data = dataset.GetReadOnlyDoubleValues(laggedVariableTreeNode.VariableName);
440            }
441            break;
442          case OpCodes.VariableCondition: {
443              var variableConditionTreeNode = (VariableConditionTreeNode)instr.dynamicNode;
444              instr.data = dataset.GetReadOnlyDoubleValues(variableConditionTreeNode.VariableName);
445            }
446            break;
447          case OpCodes.TimeLag:
448          case OpCodes.Integral:
449          case OpCodes.Derivative: {
450              var seq = GetPrefixSequence(code, i);
451              var interpreterState = new InterpreterState(seq, 0);
452              instr.data = interpreterState;
453              for (int j = 1; j != seq.Length; ++j)
454                seq[j].skip = true;
455              break;
456            }
457        }
458        #endregion
459      }
460    }
461  }
462}
Note: See TracBrowser for help on using the repository browser.