Changeset 12589

Timestamp:

07/03/15 16:59:08 (8 years ago)

Author:

gkronber

Message:

#2261 adapted interface to use IDataset instead of Dataset and added logistic regression loss function

Location:

branches/GBT-trunkintegration/HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees

Files:

• GradientBoostedTreesModel.cs (modified) (1 diff)
• LossFunctions/LogisticRegressionLoss.cs (added)
• RegressionTreeModel.cs (modified) (2 diffs)

branches/GBT-trunkintegration/HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees/GradientBoostedTreesModel.cs

@@ -39 +39 @@
     }
 
-    public IEnumerable<double> GetEstimatedValues(Dataset dataset, IEnumerable<int> rows) {
+    public IEnumerable<double> GetEstimatedValues(IDataset dataset, IEnumerable<int> rows) {
       var tuples = (from idx in Enumerable.Range(0, models.Count)
                     let model = models[idx]

branches/GBT-trunkintegration/HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees/RegressionTreeModel.cs

@@ -56 +56 @@
     }
 
-    private static double GetPredictionForRow(TreeNode[] t, int nodeIdx, Dataset ds, int row) {
+    private static double GetPredictionForRow(TreeNode[] t, int nodeIdx, IDataset ds, int row) {
       var node = t[nodeIdx];
       if (node.varName == TreeNode.NO_VARIABLE)
@@ -70 +70 @@
     }
 
-    public IEnumerable<double> GetEstimatedValues(Dataset ds, IEnumerable<int> rows) {
+    public IEnumerable<double> GetEstimatedValues(IDataset ds, IEnumerable<int> rows) {
       return rows.Select(r => GetPredictionForRow(tree, 0, ds, r));
     }