Changeset 17044 for trunk/HeuristicLab.Algorithms.DataAnalysis

Timestamp:

06/26/19 16:00:53 (5 years ago)

Author:

mkommend

Message:

#2883: Initialized Lazy object in GBTModelSurrogate.

Location:

trunk/HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees

Files:

• GradientBoostedTreesAlgorithm.cs (modified) (1 diff)
• GradientBoostedTreesModelSurrogate.cs (modified) (2 diffs)

trunk/HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees/GradientBoostedTreesAlgorithm.cs

@@ -269 +269 @@
 
         if (ModelCreation == ModelCreation.SurrogateModel) {
-          model = new GradientBoostedTreesModelSurrogate(problemData, (uint)Seed, lossFunction, Iterations, MaxSize, R, M, Nu, (GradientBoostedTreesModel)model);
+          model = new GradientBoostedTreesModelSurrogate((GradientBoostedTreesModel)model, problemData, (uint)Seed, lossFunction, Iterations, MaxSize, R, M, Nu);
         }
 

trunk/HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees/GradientBoostedTreesModelSurrogate.cs

@@ -90 +90 @@
     private Func<IGradientBoostedTreesModel> CreateLazyInitFunc(IGradientBoostedTreesModel clonedModel) {
       return () => {
-        return clonedModel == null ? RecalculateModel() : clonedModel;
+        return clonedModel ?? RecalculateModel();
       };
     }
 
     // create only the surrogate model without an actual model
-    public GradientBoostedTreesModelSurrogate(IRegressionProblemData trainingProblemData, uint seed,
+    private GradientBoostedTreesModelSurrogate(IRegressionProblemData trainingProblemData, uint seed,
       ILossFunction lossFunction, int iterations, int maxSize, double r, double m, double nu)
       : base(trainingProblemData.TargetVariable, "Gradient boosted tree model", string.Empty) {
@@ -106 +106 @@
       this.m = m;
       this.nu = nu;
+
+      actualModel = new Lazy<IGradientBoostedTreesModel>(() => RecalculateModel());
     }
 
     // wrap an actual model in a surrograte
-    public GradientBoostedTreesModelSurrogate(IRegressionProblemData trainingProblemData, uint seed,
-      ILossFunction lossFunction, int iterations, int maxSize, double r, double m, double nu,
-      IGradientBoostedTreesModel model)
+    public GradientBoostedTreesModelSurrogate(IGradientBoostedTreesModel model, IRegressionProblemData trainingProblemData, uint seed,
+      ILossFunction lossFunction, int iterations, int maxSize, double r, double m, double nu)
       : this(trainingProblemData, seed, lossFunction, iterations, maxSize, r, m, nu) {
       actualModel = new Lazy<IGradientBoostedTreesModel>(() => model);