Free cookie consent management tool by TermsFeed Policy Generator

source: stable/HeuristicLab.Algorithms.DataAnalysis.Views/3.4/GradientBoostedTreesModelEvaluationView.cs @ 17097

Last change on this file since 17097 was 17097, checked in by mkommend, 5 years ago

#2520: Merged 16565 - 16579 into stable.

File size: 2.9 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2019 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 HeuristicLab.Common;
23using HeuristicLab.MainForm;
24using HeuristicLab.Problems.DataAnalysis;
25using HeuristicLab.Problems.DataAnalysis.Views;
26
27namespace HeuristicLab.Algorithms.DataAnalysis.Views {
28  [View("GBT Model Evaluation")]
29  [Content(typeof(GradientBoostedTreesSolution), false)]
30  public partial class GradientBoostedTreesModelEvaluationView : DataAnalysisSolutionEvaluationView {
31    public new GradientBoostedTreesSolution Content {
32      get { return (GradientBoostedTreesSolution)base.Content; }
33      set { base.Content = value; }
34    }
35
36    protected override void SetEnabledStateOfControls() {
37      base.SetEnabledStateOfControls();
38      listBox.Enabled = Content != null;
39      viewHost.Enabled = Content != null;
40    }
41
42    public GradientBoostedTreesModelEvaluationView()
43      : base() {
44      InitializeComponent();
45    }
46
47    protected override void OnContentChanged() {
48      base.OnContentChanged();
49      if (Content == null) {
50        viewHost.Content = null;
51        listBox.Items.Clear();
52      } else {
53        viewHost.Content = null;
54        listBox.Items.Clear();
55        foreach (var e in Content.Model.Models) {
56          listBox.Items.Add(e);
57        }
58      }
59    }
60
61    private void listBox_SelectedIndexChanged(object sender, System.EventArgs e) {
62      var model = listBox.SelectedItem;
63      if (model == null) viewHost.Content = null;
64      else {
65        viewHost.Content = ConvertModel(model);
66      }
67    }
68
69    private void listBox_DoubleClick(object sender, System.EventArgs e) {
70      var selectedItem = listBox.SelectedItem;
71      if (selectedItem == null) return;
72      MainFormManager.MainForm.ShowContent(ConvertModel(selectedItem));
73    }
74
75    private IContent ConvertModel(object model) {
76      var treeModel = model as RegressionTreeModel;
77      if (treeModel != null)
78        return treeModel.CreateSymbolicRegressionSolution(Content.ProblemData);
79      else {
80        var regModel = model as IRegressionModel;
81        return regModel;
82      }
83    }
84  }
85}
Note: See TracBrowser for help on using the repository browser.