Add ClassificationBinomialGradientBoostLearner

mdabros · Feb 17, 2025 · f9e7085 · f9e7085
1 parent 2c34a9c
commit f9e7085
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 65 deletions.
diff --git a/src/SharpLearning.Benchmarks/Benchmarks.ClassificationLearners.cs b/src/SharpLearning.Benchmarks/Benchmarks.ClassificationLearners.cs
@@ -2,6 +2,7 @@
 using SharpLearning.AdaBoost.Learners;
 using SharpLearning.Containers.Matrices;
 using SharpLearning.DecisionTrees.Learners;
+using SharpLearning.GradientBoost.Learners;
 using SharpLearning.RandomForest.Learners;
 
 namespace SharpLearning.Benchmarks;
@@ -23,6 +24,7 @@ public class ClassificationLearners
         readonly ClassificationAdaBoostLearner m_classificationAdaBoostLearner = new();
         readonly ClassificationRandomForestLearner m_classificationRandomForestLearner = new();
         readonly ClassificationExtremelyRandomizedTreesLearner m_classificationExtremelyRandomizedTreesLearner = new();
+        readonly ClassificationBinomialGradientBoostLearner m_classificationBinomialGradientBoostLearner = new();
 
         [GlobalSetup]
         public void GlobalSetup()
@@ -57,5 +59,11 @@ public void ClassificationExtremelyRandomizedTreesLearner_Learn()
         {
             m_classificationExtremelyRandomizedTreesLearner.Learn(m_features, m_targets);
         }
+
+        [Benchmark]
+        public void ClassificationBinomialGradientBoostLearner_Learn()
+        {
+            m_classificationBinomialGradientBoostLearner.Learn(m_features, m_targets);
+        }
     }
 }
diff --git a/src/SharpLearning.Benchmarks/Benchmarks.Regression.cs b/src/SharpLearning.Benchmarks/Benchmarks.Regression.cs
diff --git a/src/SharpLearning.Benchmarks/RegressionLearnerBenchmark.cs b/src/SharpLearning.Benchmarks/RegressionLearnerBenchmark.cs