Skip to content

Commit b1576e1

Browse files
committed
Add remaining gradient boost learners
1 parent 01f68a4 commit b1576e1

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

src/SharpLearning.Benchmarks/Benchmarks.RegressionLearners.cs

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,9 @@ public class RegressionLearners
2323
readonly RegressionAdaBoostLearner m_regressionAdaBoostLearner = new();
2424
readonly RegressionRandomForestLearner m_regressionRandomForestLearner = new();
2525
readonly RegressionExtremelyRandomizedTreesLearner m_regressionExtremelyRandomizedTreesLearner = new();
26+
readonly RegressionAbsoluteLossGradientBoostLearner m_regressionAbsoluteLossGradientBoostLearner = new();
27+
readonly RegressionHuberLossGradientBoostLearner m_regressionHuberLossGradientBoostLearner = new();
28+
readonly RegressionQuantileLossGradientBoostLearner m_regressionQuantileLossGradientBoostLearner = new();
2629
readonly RegressionSquareLossGradientBoostLearner m_regressionSquareLossGradientBoostLearner = new();
2730

2831
[GlobalSetup]
@@ -58,6 +61,24 @@ public void RegressionExtremelyRandomizedTreesLearner_Learn()
5861
m_regressionExtremelyRandomizedTreesLearner.Learn(m_features, m_targets);
5962
}
6063

64+
[Benchmark]
65+
public void RegressionAbsoluteLossGradientBoostLearner_Learn()
66+
{
67+
m_regressionAbsoluteLossGradientBoostLearner.Learn(m_features, m_targets);
68+
}
69+
70+
[Benchmark]
71+
public void RegressionHuberLossGradientBoostLearner_Learn()
72+
{
73+
m_regressionHuberLossGradientBoostLearner.Learn(m_features, m_targets);
74+
}
75+
76+
[Benchmark]
77+
public void RegressionQuantileLossGradientBoostLearner_Learn()
78+
{
79+
m_regressionQuantileLossGradientBoostLearner.Learn(m_features, m_targets);
80+
}
81+
6182
[Benchmark]
6283
public void RegressionSquareLossGradientBoostLearner_Learn()
6384
{

0 commit comments

Comments
 (0)