[8578] | 1 | #region License Information
|
---|
| 2 | /* HeuristicLab
|
---|
[13082] | 3 | * Copyright (C) 2002-2015 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
|
---|
[8578] | 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 |
|
---|
| 21 | #endregion
|
---|
| 22 |
|
---|
| 23 | using System;
|
---|
| 24 | using System.Collections.Generic;
|
---|
| 25 | using HeuristicLab.PluginInfrastructure;
|
---|
| 26 |
|
---|
| 27 | namespace HeuristicLab.Problems.DataAnalysis.Views {
|
---|
| 28 | [NonDiscoverableType]
|
---|
| 29 | internal class FeatureCorrelationCache : Object {
|
---|
[8729] | 30 | private Dictionary<FeatureCorrelationEnums.CorrelationCalculators, Dictionary<FeatureCorrelationEnums.Partitions, double[,]>> correlationsCache;
|
---|
[8578] | 31 |
|
---|
| 32 | public FeatureCorrelationCache()
|
---|
| 33 | : base() {
|
---|
| 34 | InitializeCaches();
|
---|
| 35 | }
|
---|
| 36 |
|
---|
| 37 | private void InitializeCaches() {
|
---|
[8729] | 38 | correlationsCache = new Dictionary<FeatureCorrelationEnums.CorrelationCalculators, Dictionary<FeatureCorrelationEnums.Partitions, double[,]>>();
|
---|
| 39 | foreach (var calc in FeatureCorrelationEnums.EnumToList<FeatureCorrelationEnums.CorrelationCalculators>()) {
|
---|
| 40 | correlationsCache.Add(calc, new Dictionary<FeatureCorrelationEnums.Partitions, double[,]>());
|
---|
[8578] | 41 | }
|
---|
| 42 | }
|
---|
| 43 |
|
---|
| 44 | public void Reset() {
|
---|
| 45 | InitializeCaches();
|
---|
| 46 | }
|
---|
| 47 |
|
---|
[8729] | 48 | public double[,] GetCorrelation(FeatureCorrelationEnums.CorrelationCalculators calc, FeatureCorrelationEnums.Partitions partition) {
|
---|
[8578] | 49 | double[,] corr;
|
---|
| 50 | correlationsCache[calc].TryGetValue(partition, out corr);
|
---|
| 51 | return corr;
|
---|
| 52 | }
|
---|
| 53 |
|
---|
[8729] | 54 | public void SetCorrelation(FeatureCorrelationEnums.CorrelationCalculators calc, FeatureCorrelationEnums.Partitions partition, double[,] correlation) {
|
---|
[8578] | 55 | correlationsCache[calc][partition] = correlation;
|
---|
| 56 | }
|
---|
| 57 | }
|
---|
| 58 | }
|
---|