@@ -15,11 +15,11 @@ public abstract class BoostingFastTreeTrainerBase<TArgs, TTransformer, TModel> :
15
15
where TArgs : BoostedTreeArgs , new ( )
16
16
where TModel : class
17
17
{
18
- protected BoostingFastTreeTrainerBase ( IHostEnvironment env , TArgs args , SchemaShape . Column label ) : base ( env , args , label )
18
+ internal BoostingFastTreeTrainerBase ( IHostEnvironment env , TArgs args , SchemaShape . Column label ) : base ( env , args , label )
19
19
{
20
20
}
21
21
22
- protected BoostingFastTreeTrainerBase ( IHostEnvironment env ,
22
+ internal BoostingFastTreeTrainerBase ( IHostEnvironment env ,
23
23
SchemaShape . Column label ,
24
24
string featureColumn ,
25
25
string weightColumn ,
@@ -33,7 +33,7 @@ protected BoostingFastTreeTrainerBase(IHostEnvironment env,
33
33
Args . LearningRates = learningRate ;
34
34
}
35
35
36
- protected override void CheckArgs ( IChannel ch )
36
+ internal override void CheckArgs ( IChannel ch )
37
37
{
38
38
if ( Args . OptimizationAlgorithm == BoostedTreeArgs . OptimizationAlgorithmType . AcceleratedGradientDescent )
39
39
Args . UseLineSearch = true ;
@@ -116,7 +116,7 @@ internal override IGradientAdjuster MakeGradientWrapper(IChannel ch)
116
116
return new BestStepRegressionGradientWrapper ( ) ;
117
117
}
118
118
119
- protected override bool ShouldStop ( IChannel ch , ref IEarlyStoppingCriterion earlyStoppingRule , ref int bestIteration )
119
+ internal override bool ShouldStop ( IChannel ch , ref IEarlyStoppingCriterion earlyStoppingRule , ref int bestIteration )
120
120
{
121
121
if ( Args . EarlyStoppingRule == null )
122
122
return false ;
@@ -148,7 +148,7 @@ protected override bool ShouldStop(IChannel ch, ref IEarlyStoppingCriterion earl
148
148
return shouldStop ;
149
149
}
150
150
151
- protected override int GetBestIteration ( IChannel ch )
151
+ internal override int GetBestIteration ( IChannel ch )
152
152
{
153
153
int bestIteration = Ensemble . NumTrees ;
154
154
if ( ! Args . WriteLastEnsemble && PruningTest != null )
@@ -162,15 +162,15 @@ protected override int GetBestIteration(IChannel ch)
162
162
/// <summary>
163
163
/// Retrieves max tree output if best regression step option is active or returns negative value otherwise.
164
164
/// </summary>
165
- protected double BsrMaxTreeOutput ( )
165
+ internal double BsrMaxTreeOutput ( )
166
166
{
167
167
if ( Args . BestStepRankingRegressionTrees )
168
168
return Args . MaxTreeOutput ;
169
169
else
170
170
return - 1 ;
171
171
}
172
172
173
- protected override bool ShouldRandomStartOptimizer ( )
173
+ internal override bool ShouldRandomStartOptimizer ( )
174
174
{
175
175
return Args . RandomStart ;
176
176
}
0 commit comments