Medical Imaging Interaction Toolkit  2022.04.99-194dd5d8
Medical Imaging Interaction Toolkit
mitkLRDensityEstimation.h
Go to the documentation of this file.
1 /*============================================================================
2 
3 The Medical Imaging Interaction Toolkit (MITK)
4 
5 Copyright (c) German Cancer Research Center (DKFZ)
6 All rights reserved.
7 
8 Use of this source code is governed by a 3-clause BSD license that can be
9 found in the LICENSE file.
10 
11 ============================================================================*/
12 
13 #ifndef mitkLRDensityEstimation_h
14 #define mitkLRDensityEstimation_h
15 
16 #include <vigra/matrix.hxx>
17 #include <vigra/random_forest.hxx>
18 
19 #include <mitkDataCollection.h>
20 
22 
23 namespace mitk
24 {
26  {
27  public:
28  typedef vigra::MultiArray<2, double> MatrixType;
29 
30  void SetCollection(DataCollection::Pointer data);
31  DataCollection::Pointer GetCollection();
32 
33  void SetTestMask(std::string name);
34  std::string GetTestMask();
35 
36  void SetTrainMask(std::string name);
37  std::string GetTrainMask();
38 
39  void SetWeightName(std::string name);
40  std::string GetWeightName();
41 
42  void Update();
44  void SetModalities(std::vector<std::string> modalities);
45  std::vector<std::string> GetModalities();
46 
47  void Predict();
48 
49  private:
50  DataCollection::Pointer m_Collection;
51  MatrixType m_Weights;
52  std::string m_TestMask;
53  std::string m_TrainMask;
54  std::string m_WeightName;
55  std::vector<std::string> m_Modalities;
56 
57  std::string m_WeightFileName;
58  }; // class LRDensityEstimation
59 } //namespace mitk
60 
61 #endif //mitkLRDensityEstimation_h
Follow Up Storage - Class to facilitate loading/accessing structured follow-up data.
Definition: testcase.h:28
vigra::MultiArray< 2, double > MatrixType
DataCollection - Class to facilitate loading/accessing structured data.
#define MITKCLIMPORTANCEWEIGHTING_EXPORT
itk::SmartPointer< Self > Pointer
Definition: mitkBaseData.h:46