Free cookie consent management tool by TermsFeed Policy Generator

source: branches/2521_ProblemRefactoring/HeuristicLab.Problems.Instances.DataAnalysis/3.3/Regression/Feynman/FeynmanBonus5.cs @ 18086

Last change on this file since 18086 was 18086, checked in by mkommend, 2 years ago

#2521: Merged trunk changes into branch.

File size: 2.9 KB
Line 
1using System;
2using System.Collections.Generic;
3using System.Linq;
4using HeuristicLab.Common;
5using HeuristicLab.Random;
6
7namespace HeuristicLab.Problems.Instances.DataAnalysis {
8  public class FeynmanBonus5 : FeynmanDescriptor {
9    private readonly int testSamples;
10    private readonly int trainingSamples;
11
12    public FeynmanBonus5() : this((int) DateTime.Now.Ticks, 10000, 10000, null) { }
13
14    public FeynmanBonus5(int seed) {
15      Seed            = seed;
16      trainingSamples = 10000;
17      testSamples     = 10000;
18      noiseRatio      = null;
19    }
20
21    public FeynmanBonus5(int seed, int trainingSamples, int testSamples, double? noiseRatio) {
22      Seed                 = seed;
23      this.trainingSamples = trainingSamples;
24      this.testSamples     = testSamples;
25      this.noiseRatio      = noiseRatio;
26    }
27
28    public override string Name {
29      get {
30        return string.Format(
31          "Relativistic aberation: arccos((cos(theta2)-v/c)/(1-v/c*cos(theta2))) | {0}",
32          noiseRatio == null ? "no noise" : string.Format(System.Globalization.CultureInfo.InvariantCulture, "noise={0:g}",noiseRatio));
33      }
34    }
35
36    protected override string TargetVariable { get { return noiseRatio == null ? "theta1" : "theta1_noise"; } }
37
38    protected override string[] VariableNames {
39      get { return noiseRatio == null ? new[] { "c", "v", "theta2", "theta1" } : new[] { "c", "v", "theta2", "theta1", "theta1_noise" }; }
40    }
41
42    protected override string[] AllowedInputVariables { get { return new[] {"c", "v", "theta2"}; } }
43
44    public int Seed { get; private set; }
45
46    protected override int TrainingPartitionStart { get { return 0; } }
47    protected override int TrainingPartitionEnd { get { return trainingSamples; } }
48    protected override int TestPartitionStart { get { return trainingSamples; } }
49    protected override int TestPartitionEnd { get { return trainingSamples + testSamples; } }
50
51    protected override List<List<double>> GenerateValues() {
52      var rand = new MersenneTwister((uint) Seed);
53
54      var data   = new List<List<double>>();
55      var c      = ValueGenerator.GenerateUniformDistributedValues(rand.Next(), TestPartitionEnd, 4, 6).ToList();
56      var v      = ValueGenerator.GenerateUniformDistributedValues(rand.Next(), TestPartitionEnd, 1, 3).ToList();
57      var theta2 = ValueGenerator.GenerateUniformDistributedValues(rand.Next(), TestPartitionEnd, 1, 3).ToList();
58
59      var theta1 = new List<double>();
60
61      data.Add(c);
62      data.Add(v);
63      data.Add(theta2);
64      data.Add(theta1);
65
66      for (var i = 0; i < c.Count; i++) {
67        var res = Math.Acos((Math.Cos(theta2[i]) - v[i] / c[i]) / (1 - v[i] / c[i] * Math.Cos(theta2[i])));
68        theta1.Add(res);
69      }
70
71      var targetNoise = ValueGenerator.GenerateNoise(theta1, rand, noiseRatio);
72      if (targetNoise != null) data.Add(targetNoise);
73
74      return data;
75    }
76  }
77}
Note: See TracBrowser for help on using the repository browser.