Free cookie consent management tool by TermsFeed Policy Generator

source: branches/2892_LR-prediction-intervals/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/DataAnalysisProblemData.cs @ 16388

Last change on this file since 16388 was 16388, checked in by gkronber, 5 years ago

#2892: Merging r15750:16382 (HEAD) from trunk to branch, resolving conflicts

File size: 9.4 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 System.Text;
26using HeuristicLab.Collections;
27using HeuristicLab.Common;
28using HeuristicLab.Core;
29using HeuristicLab.Data;
30using HeuristicLab.Parameters;
31using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
32
33namespace HeuristicLab.Problems.DataAnalysis {
34  [StorableClass]
35  public abstract class DataAnalysisProblemData : ParameterizedNamedItem, IDataAnalysisProblemData {
36    protected const string DatasetParameterName = "Dataset";
37    protected const string InputVariablesParameterName = "InputVariables";
38    protected const string TrainingPartitionParameterName = "TrainingPartition";
39    protected const string TestPartitionParameterName = "TestPartition";
40    protected const string TransformationsParameterName = "Transformations";
41
42    #region parameter properites
43    //mkommend: inserted parameter caching due to performance reasons
44    private IFixedValueParameter<Dataset> datasetParameter;
45    public IFixedValueParameter<Dataset> DatasetParameter {
46      get {
47        if (datasetParameter == null) datasetParameter = (IFixedValueParameter<Dataset>)Parameters[DatasetParameterName];
48        return datasetParameter;
49      }
50    }
51
52    private IFixedValueParameter<ReadOnlyCheckedItemList<StringValue>> inputVariablesParameter;
53    public IFixedValueParameter<ReadOnlyCheckedItemList<StringValue>> InputVariablesParameter {
54      get {
55        if (inputVariablesParameter == null) inputVariablesParameter = (IFixedValueParameter<ReadOnlyCheckedItemList<StringValue>>)Parameters[InputVariablesParameterName];
56        return inputVariablesParameter;
57      }
58    }
59
60    private IFixedValueParameter<IntRange> trainingPartitionParameter;
61    public IFixedValueParameter<IntRange> TrainingPartitionParameter {
62      get {
63        if (trainingPartitionParameter == null) trainingPartitionParameter = (IFixedValueParameter<IntRange>)Parameters[TrainingPartitionParameterName];
64        return trainingPartitionParameter;
65      }
66    }
67
68    private IFixedValueParameter<IntRange> testPartitionParameter;
69    public IFixedValueParameter<IntRange> TestPartitionParameter {
70      get {
71        if (testPartitionParameter == null) testPartitionParameter = (IFixedValueParameter<IntRange>)Parameters[TestPartitionParameterName];
72        return testPartitionParameter;
73      }
74    }
75
76    public IFixedValueParameter<ReadOnlyItemList<ITransformation>> TransformationsParameter {
77      get { return (IFixedValueParameter<ReadOnlyItemList<ITransformation>>)Parameters[TransformationsParameterName]; }
78    }
79    #endregion
80
81    #region properties
82    protected bool isEmpty = false;
83    public bool IsEmpty {
84      get { return isEmpty; }
85    }
86    public IDataset Dataset {
87      get { return DatasetParameter.Value; }
88    }
89    public ICheckedItemList<StringValue> InputVariables {
90      get { return InputVariablesParameter.Value; }
91    }
92    public IEnumerable<string> AllowedInputVariables {
93      get { return InputVariables.CheckedItems.Select(x => x.Value.Value); }
94    }
95
96    public double[,] AllowedInputsTrainingValues {
97      get { return Dataset.ToArray(AllowedInputVariables, TrainingIndices); }
98    }
99
100    public double[,] AllowedInputsTestValues { get { return Dataset.ToArray(AllowedInputVariables, TestIndices); } }
101    public IntRange TrainingPartition {
102      get { return TrainingPartitionParameter.Value; }
103    }
104    public IntRange TestPartition {
105      get { return TestPartitionParameter.Value; }
106    }
107
108    public virtual IEnumerable<int> AllIndices {
109      get { return Enumerable.Range(0, Dataset.Rows); }
110    }
111    public virtual IEnumerable<int> TrainingIndices {
112      get {
113        return Enumerable.Range(TrainingPartition.Start, Math.Max(0, TrainingPartition.End - TrainingPartition.Start))
114                         .Where(IsTrainingSample);
115      }
116    }
117    public virtual IEnumerable<int> TestIndices {
118      get {
119        return Enumerable.Range(TestPartition.Start, Math.Max(0, TestPartition.End - TestPartition.Start))
120           .Where(IsTestSample);
121      }
122    }
123
124    public IEnumerable<ITransformation> Transformations {
125      get { return TransformationsParameter.Value; }
126    }
127
128    public virtual bool IsTrainingSample(int index) {
129      return index >= 0 && index < Dataset.Rows &&
130             TrainingPartition.Start <= index && index < TrainingPartition.End &&
131             (index < TestPartition.Start || TestPartition.End <= index);
132    }
133
134    public virtual bool IsTestSample(int index) {
135      return index >= 0 && index < Dataset.Rows &&
136             TestPartition.Start <= index && index < TestPartition.End;
137    }
138    #endregion
139
140    protected DataAnalysisProblemData(DataAnalysisProblemData original, Cloner cloner)
141      : base(original, cloner) {
142      isEmpty = original.isEmpty;
143      RegisterEventHandlers();
144    }
145    [StorableConstructor]
146    protected DataAnalysisProblemData(bool deserializing) : base(deserializing) { }
147
148    [StorableHook(HookType.AfterDeserialization)]
149    private void AfterDeserialization() {
150      if (!Parameters.ContainsKey(TransformationsParameterName)) {
151        Parameters.Add(new FixedValueParameter<ReadOnlyItemList<ITransformation>>(TransformationsParameterName, "", new ItemList<ITransformation>().AsReadOnly()));
152        TransformationsParameter.Hidden = true;
153      }
154      RegisterEventHandlers();
155    }
156
157    protected DataAnalysisProblemData(IDataset dataset, IEnumerable<string> allowedInputVariables, IEnumerable<ITransformation> transformations = null) {
158      if (dataset == null) throw new ArgumentNullException("The dataset must not be null.");
159      if (allowedInputVariables == null) throw new ArgumentNullException("The allowed input variables must not be null.");
160
161      if (allowedInputVariables.Except(dataset.DoubleVariables).Except(dataset.StringVariables).Any())
162        throw new ArgumentException("All allowed input variables must be present in the dataset and of type double or string.");
163
164      var variables = dataset.VariableNames.Where(variable => dataset.VariableHasType<double>(variable) || dataset.VariableHasType<string>(variable));
165      var inputVariables = new CheckedItemList<StringValue>(variables.Select(x => new StringValue(x).AsReadOnly()));
166      foreach (StringValue x in inputVariables)
167        inputVariables.SetItemCheckedState(x, allowedInputVariables.Contains(x.Value));
168
169      int trainingPartitionStart = 0;
170      int trainingPartitionEnd = dataset.Rows / 2;
171      int testPartitionStart = dataset.Rows / 2;
172      int testPartitionEnd = dataset.Rows;
173
174      var transformationsList = new ItemList<ITransformation>(transformations ?? Enumerable.Empty<ITransformation>());
175
176      Parameters.Add(new FixedValueParameter<Dataset>(DatasetParameterName, "", (Dataset)dataset));
177      Parameters.Add(new FixedValueParameter<ReadOnlyCheckedItemList<StringValue>>(InputVariablesParameterName, "", inputVariables.AsReadOnly()));
178      Parameters.Add(new FixedValueParameter<IntRange>(TrainingPartitionParameterName, "", new IntRange(trainingPartitionStart, trainingPartitionEnd)));
179      Parameters.Add(new FixedValueParameter<IntRange>(TestPartitionParameterName, "", new IntRange(testPartitionStart, testPartitionEnd)));
180      Parameters.Add(new FixedValueParameter<ReadOnlyItemList<ITransformation>>(TransformationsParameterName, "", transformationsList.AsReadOnly()));
181
182      TransformationsParameter.Hidden = true;
183
184      ((ValueParameter<Dataset>)DatasetParameter).ReactOnValueToStringChangedAndValueItemImageChanged = false;
185      RegisterEventHandlers();
186    }
187
188    private void RegisterEventHandlers() {
189      DatasetParameter.ValueChanged += new EventHandler(Parameter_ValueChanged);
190      InputVariables.CheckedItemsChanged += new CollectionItemsChangedEventHandler<IndexedItem<StringValue>>(InputVariables_CheckedItemsChanged);
191      TrainingPartition.ValueChanged += new EventHandler(Parameter_ValueChanged);
192      TestPartition.ValueChanged += new EventHandler(Parameter_ValueChanged);
193      TransformationsParameter.ValueChanged += new EventHandler(Parameter_ValueChanged);
194    }
195
196    private void InputVariables_CheckedItemsChanged(object sender, CollectionItemsChangedEventArgs<IndexedItem<StringValue>> e) {
197      OnChanged();
198    }
199
200    private void Parameter_ValueChanged(object sender, EventArgs e) {
201      OnChanged();
202    }
203
204    public event EventHandler Changed;
205    protected virtual void OnChanged() {
206      var listeners = Changed;
207      if (listeners != null) listeners(this, EventArgs.Empty);
208    }
209  }
210}
Note: See TracBrowser for help on using the repository browser.