#region License Information /* HeuristicLab * Copyright (C) 2002-2016 Heuristic and Evolutionary Algorithms Laboratory (HEAL) * * This file is part of HeuristicLab. * * HeuristicLab is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * HeuristicLab is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with HeuristicLab. If not, see . */ #endregion using HeuristicLab.Common; using HeuristicLab.Core; using HeuristicLab.Data; using HeuristicLab.Operators; using HeuristicLab.Optimization; using HeuristicLab.Parameters; using HeuristicLab.Persistence.Default.CompositeSerializers.Storable; using HeuristicLab.Problems.DataAnalysis; namespace HeuristicLab.Algorithms.DataAnalysis { [StorableClass] [Item(Name = "GaussianProcessClassificationSolutionCreator", Description = "Creates a Gaussian process solution from a trained model.")] public sealed class GaussianProcessClassificationSolutionCreator : SingleSuccessorOperator { private const string ProblemDataParameterName = "ProblemData"; private const string ModelParameterName = "GaussianProcessClassificationModel"; private const string SolutionParameterName = "Solution"; private const string ResultsParameterName = "Results"; private const string TrainingAccuracyResultName = "Accuracy (training)"; private const string TestAccuracyResultName = "Accuracy (test)"; private const string CreateSolutionParameterName = "CreateSolution"; #region Parameter Properties public ILookupParameter ProblemDataParameter { get { return (ILookupParameter)Parameters[ProblemDataParameterName]; } } public ILookupParameter SolutionParameter { get { return (ILookupParameter)Parameters[SolutionParameterName]; } } public ILookupParameter ModelParameter { get { return (ILookupParameter)Parameters[ModelParameterName]; } } public ILookupParameter ResultsParameter { get { return (ILookupParameter)Parameters[ResultsParameterName]; } } public ILookupParameter CreateSolutionParameter { get { return (ILookupParameter)Parameters[CreateSolutionParameterName]; } } #endregion [StorableConstructor] private GaussianProcessClassificationSolutionCreator(bool deserializing) : base(deserializing) { } private GaussianProcessClassificationSolutionCreator(GaussianProcessClassificationSolutionCreator original, Cloner cloner) : base(original, cloner) { } public GaussianProcessClassificationSolutionCreator() : base() { // in Parameters.Add(new LookupParameter(ProblemDataParameterName, "The classification problem data for the Gaussian process solution.")); Parameters.Add(new LookupParameter(ModelParameterName, "The Gaussian process classification model to use for the solution.")); Parameters.Add(new LookupParameter(CreateSolutionParameterName, "Flag that indicates if a solution should be produced at the end of the run")); // in & out Parameters.Add(new LookupParameter(ResultsParameterName, "The result collection of the algorithm.")); // out Parameters.Add(new LookupParameter(SolutionParameterName, "The produced Gaussian process solution.")); } [StorableHook(HookType.AfterDeserialization)] private void AfterDeserialization() { // BackwardsCompatibility3.3 #region Backwards compatible code, remove with 3.4 if (!Parameters.ContainsKey(CreateSolutionParameterName)) { Parameters.Add(new LookupParameter(CreateSolutionParameterName, "Flag that indicates if a solution should be produced at the end of the run")); } #endregion } public override IDeepCloneable Clone(Cloner cloner) { return new GaussianProcessClassificationSolutionCreator(this, cloner); } public override IOperation Apply() { if (ModelParameter.ActualValue != null && CreateSolutionParameter.ActualValue.Value == true) { var m = (IGaussianProcessModel)ModelParameter.ActualValue.Clone(); m.FixParameters(); var data = (IClassificationProblemData)ProblemDataParameter.ActualValue.Clone(); var model = new DiscriminantFunctionClassificationModel(m, new NormalDistributionCutPointsThresholdCalculator()); model.RecalculateModelParameters(data, data.TrainingIndices); var s = model.CreateDiscriminantFunctionClassificationSolution(data); SolutionParameter.ActualValue = s; var results = ResultsParameter.ActualValue; if (!results.ContainsKey(SolutionParameterName)) { results.Add(new Result(SolutionParameterName, "The Gaussian process classification solution", s)); results.Add(new Result(TrainingAccuracyResultName, "The accuracy of the Gaussian process solution on the training partition.", new DoubleValue(s.TrainingAccuracy))); results.Add(new Result(TestAccuracyResultName, "The accuracy of the Gaussian process solution on the test partition.", new DoubleValue(s.TestAccuracy))); } else { results[SolutionParameterName].Value = s; results[TrainingAccuracyResultName].Value = new DoubleValue(s.TrainingAccuracy); results[TestAccuracyResultName].Value = new DoubleValue(s.TestAccuracy); } } return base.Apply(); } } }