Free cookie consent management tool by TermsFeed Policy Generator

source: branches/DataAnalysis Refactoring/HeuristicLab.Problems.DataAnalysis/3.4/RegressionProblemData.cs @ 5601

Last change on this file since 5601 was 5601, checked in by mkommend, 14 years ago

#1418: Finished work on ProblemData.

File size: 5.2 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2011 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.IO;
25using System.Linq;
26using HeuristicLab.Common;
27using HeuristicLab.Core;
28using HeuristicLab.Data;
29using HeuristicLab.Parameters;
30using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
31
32namespace HeuristicLab.Problems.DataAnalysis {
33  [StorableClass]
34  [Item("RegressionProblemData", "Represents an item containing all data defining a regression problem.")]
35  public sealed class RegressionProblemData : DataAnalysisProblemData, IRegressionProblemData {
36    private const string TargetVariableParameterName = "TargetVariable";
37
38    #region default data
39    private static double[,] kozaF1 = new double[,] {
40          {2.017885919, -1.449165046},
41          {1.30060506,  -1.344523885},
42          {1.147134798, -1.317989331},
43          {0.877182504, -1.266142284},
44          {0.852562452, -1.261020794},
45          {0.431095788, -1.158793317},
46          {0.112586002, -1.050908405},
47          {0.04594507,  -1.021989402},
48          {0.042572879, -1.020438113},
49          {-0.074027291,  -0.959859562},
50          {-0.109178553,  -0.938094706},
51          {-0.259721109,  -0.803635355},
52          {-0.272991057,  -0.387519561},
53          {-0.161978191,  -0.193611001},
54          {-0.102489983,  -0.114215349},
55          {-0.01469968, -0.014918985},
56          {-0.008863365,  -0.008942626},
57          {0.026751057, 0.026054094},
58          {0.166922436, 0.14309643},
59          {0.176953808, 0.1504144},
60          {0.190233418, 0.159916534},
61          {0.199800708, 0.166635331},
62          {0.261502822, 0.207600348},
63          {0.30182879,  0.232370249},
64          {0.83763905,  0.468046718}
65    };
66    private static Dataset defaultDataset;
67    private static IEnumerable<string> defaultAllowedInputVariables;
68    private static string defaultTargetVariable;
69
70    static RegressionProblemData() {
71      defaultDataset = new Dataset(new string[] { "y", "x" }, kozaF1);
72      defaultDataset.Name = "Fourth-order Polynomial Function Benchmark Dataset";
73      defaultDataset.Description = "f(x) = x^4 + x^3 + x^2 + x^1";
74      defaultAllowedInputVariables = new List<string>() { "x" };
75      defaultTargetVariable = "y";
76    }
77    #endregion
78
79    public IValueParameter<StringValue> TargetVariableParameter {
80      get { return (IValueParameter<StringValue>)Parameters[TargetVariableParameterName]; }
81    }
82    public string TargetVariable {
83      get { return TargetVariableParameter.Value.Value; }
84    }
85
86    [StorableConstructor]
87    private RegressionProblemData(bool deserializing) : base(deserializing) { }
88    [StorableHook(HookType.AfterDeserialization)]
89    private void AfterDeserialization() {
90      RegisterParameterEvents();
91    }
92
93
94    private RegressionProblemData(RegressionProblemData original, Cloner cloner)
95      : base(original, cloner) {
96      RegisterParameterEvents();
97    }
98    public override IDeepCloneable Clone(Cloner cloner) { return new RegressionProblemData(this, cloner); }
99
100    public RegressionProblemData()
101      : this(defaultDataset, defaultAllowedInputVariables, defaultTargetVariable) {
102    }
103
104    public RegressionProblemData(Dataset dataset, IEnumerable<string> allowedInputVariables, string targetVariable)
105      : base(dataset, allowedInputVariables) {
106      var variables = InputVariables.Select(x => x.AsReadOnly()).ToList();
107      Parameters.Add(new ConstrainedValueParameter<StringValue>(TargetVariableParameterName, new ItemSet<StringValue>(variables), variables.Where(x => x.Value == targetVariable).First()));
108    }
109
110    private void RegisterParameterEvents() {
111      TargetVariableParameter.ValueChanged += new EventHandler(TargetVariableParameter_ValueChanged);
112    }
113    private void TargetVariableParameter_ValueChanged(object sender, EventArgs e) {
114      OnChanged();
115    }
116
117    #region Import from file
118    public static RegressionProblemData ImportFromFile(string fileName) {
119      TableFileParser csvFileParser = new TableFileParser();
120      csvFileParser.Parse(fileName);
121
122      Dataset dataset = new Dataset(csvFileParser.VariableNames, csvFileParser.Values);
123      dataset.Name = Path.GetFileName(fileName);
124
125      RegressionProblemData problemData = new RegressionProblemData(dataset, dataset.VariableNames.Skip(1), dataset.VariableNames.First());
126      problemData.Name = "Data imported from " + Path.GetFileName(fileName);
127      return problemData;
128    }
129    #endregion
130  }
131}
Note: See TracBrowser for help on using the repository browser.