Free cookie consent management tool by TermsFeed Policy Generator

source: trunk/sources/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/DataAnalysisProblemData.cs @ 6666

Last change on this file since 6666 was 6666, checked in by mkommend, 11 years ago

#1592: Enabled creation of empty ensemble solutions and problem data changes.

File size: 6.4 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.Linq;
25using HeuristicLab.Collections;
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  public abstract class DataAnalysisProblemData : ParameterizedNamedItem, IDataAnalysisProblemData {
35    protected const string DatasetParameterName = "Dataset";
36    protected const string InputVariablesParameterName = "InputVariables";
37    protected const string TrainingPartitionParameterName = "TrainingPartition";
38    protected const string TestPartitionParameterName = "TestPartition";
39
40    #region parameter properites
41    public IFixedValueParameter<Dataset> DatasetParameter {
42      get { return (IFixedValueParameter<Dataset>)Parameters[DatasetParameterName]; }
43    }
44    public IFixedValueParameter<ReadOnlyCheckedItemList<StringValue>> InputVariablesParameter {
45      get { return (IFixedValueParameter<ReadOnlyCheckedItemList<StringValue>>)Parameters[InputVariablesParameterName]; }
46    }
47    public IFixedValueParameter<IntRange> TrainingPartitionParameter {
48      get { return (IFixedValueParameter<IntRange>)Parameters[TrainingPartitionParameterName]; }
49    }
50    public IFixedValueParameter<IntRange> TestPartitionParameter {
51      get { return (IFixedValueParameter<IntRange>)Parameters[TestPartitionParameterName]; }
52    }
53    #endregion
54
55    #region properties
56    protected bool isEmpty = false;
57    public bool IsEmpty {
58      get { return isEmpty; }
59    }
60    public Dataset Dataset {
61      get { return DatasetParameter.Value; }
62    }
63    public ICheckedItemList<StringValue> InputVariables {
64      get { return InputVariablesParameter.Value; }
65    }
66    public IEnumerable<string> AllowedInputVariables {
67      get { return InputVariables.CheckedItems.Select(x => x.Value.Value); }
68    }
69
70    public IntRange TrainingPartition {
71      get { return TrainingPartitionParameter.Value; }
72    }
73    public IntRange TestPartition {
74      get { return TestPartitionParameter.Value; }
75    }
76
77    public virtual IEnumerable<int> TrainingIndizes {
78      get {
79        return Enumerable.Range(TrainingPartition.Start, TrainingPartition.End - TrainingPartition.Start)
80                         .Where(i => i >= 0 && i < Dataset.Rows && (i < TestPartition.Start || TestPartition.End <= i));
81      }
82    }
83    public virtual IEnumerable<int> TestIndizes {
84      get {
85        return Enumerable.Range(TestPartition.Start, TestPartition.End - TestPartition.Start)
86           .Where(i => i >= 0 && i < Dataset.Rows);
87      }
88    }
89    #endregion
90
91    protected DataAnalysisProblemData(DataAnalysisProblemData original, Cloner cloner)
92      : base(original, cloner) {
93      isEmpty = original.isEmpty;
94      RegisterEventHandlers();
95    }
96    [StorableConstructor]
97    protected DataAnalysisProblemData(bool deserializing) : base(deserializing) { }
98    [StorableHook(HookType.AfterDeserialization)]
99    private void AfterDeserialization() {
100      RegisterEventHandlers();
101    }
102
103    protected DataAnalysisProblemData(Dataset dataset, IEnumerable<string> allowedInputVariables) {
104      if (dataset == null) throw new ArgumentNullException("The dataset must not be null.");
105      if (allowedInputVariables == null) throw new ArgumentNullException("The allowedInputVariables must not be null.");
106
107      if (allowedInputVariables.Except(dataset.VariableNames).Any())
108        throw new ArgumentException("All allowed input variables must be present in the dataset.");
109
110      var inputVariables = new CheckedItemList<StringValue>(dataset.VariableNames.Select(x => new StringValue(x)));
111      foreach (StringValue x in inputVariables)
112        inputVariables.SetItemCheckedState(x, allowedInputVariables.Contains(x.Value));
113
114      int trainingPartitionStart = 0;
115      int trainingPartitionEnd = dataset.Rows / 2;
116      int testPartitionStart = dataset.Rows / 2;
117      int testPartitionEnd = dataset.Rows;
118
119      Parameters.Add(new FixedValueParameter<Dataset>(DatasetParameterName, "", dataset));
120      Parameters.Add(new FixedValueParameter<ReadOnlyCheckedItemList<StringValue>>(InputVariablesParameterName, "", inputVariables.AsReadOnly()));
121      Parameters.Add(new FixedValueParameter<IntRange>(TrainingPartitionParameterName, "", new IntRange(trainingPartitionStart, trainingPartitionEnd)));
122      Parameters.Add(new FixedValueParameter<IntRange>(TestPartitionParameterName, "", new IntRange(testPartitionStart, testPartitionEnd)));
123
124      ((ValueParameter<Dataset>)DatasetParameter).ReactOnValueToStringChangedAndValueItemImageChanged = false;
125      RegisterEventHandlers();
126    }
127
128    private void RegisterEventHandlers() {
129      DatasetParameter.ValueChanged += new EventHandler(Parameter_ValueChanged);
130      InputVariables.CheckedItemsChanged += new CollectionItemsChangedEventHandler<IndexedItem<StringValue>>(InputVariables_CheckedItemsChanged);
131      TrainingPartition.ValueChanged += new EventHandler(Parameter_ValueChanged);
132      TestPartition.ValueChanged += new EventHandler(Parameter_ValueChanged);
133    }
134
135    private void InputVariables_CheckedItemsChanged(object sender, CollectionItemsChangedEventArgs<IndexedItem<StringValue>> e) {
136      OnChanged();
137    }
138
139    private void Parameter_ValueChanged(object sender, EventArgs e) {
140      OnChanged();
141    }
142
143    public event EventHandler Changed;
144    protected virtual void OnChanged() {
145      var listeners = Changed;
146      if (listeners != null) listeners(this, EventArgs.Empty);
147    }
148  }
149}
Note: See TracBrowser for help on using the repository browser.