Skip to content

Commit beb322f

Browse files
committed
Some docs
1 parent 1fb7650 commit beb322f

File tree

1 file changed

+33
-1
lines changed

1 file changed

+33
-1
lines changed

src/Microsoft.ML.FastTree/FastTree.cs

+33-1
Original file line numberDiff line numberDiff line change
@@ -2813,7 +2813,7 @@ public abstract class TreeEnsembleModelParameters :
28132813
{
28142814
// The below two properties are necessary for tree Visualizer
28152815
[BestFriend]
2816-
internal InternalTreeEnsemble TrainedEnsemble;
2816+
internal InternalTreeEnsemble TrainedEnsemble { get; }
28172817

28182818
int ITreeEnsemble.NumTrees => TrainedEnsemble.NumTrees;
28192819

@@ -2919,6 +2919,12 @@ protected TreeEnsembleModelParameters(IHostEnvironment env, string name, ModelLo
29192919
OutputType = NumberType.Float;
29202920
}
29212921

2922+
/// <summary>
2923+
/// This function should be implemented in derived classs to create strongly-typed TreeEnsemble
2924+
/// from <see cref="TrainedEnsemble"/> and possibly other internal attributes in
2925+
/// <see cref="TreeEnsembleModelParameters"/>. This also implies we always call this function
2926+
/// after initializing <see cref="TrainedEnsemble"/>.
2927+
/// </summary>
29222928
protected abstract void CreateTreeEnsembleFromInternalDataStructure();
29232929

29242930
[BestFriend]
@@ -3382,6 +3388,16 @@ public TreeNode(Dictionary<string, object> keyValues)
33823388
}
33833389
}
33843390

3391+
/// <summary>
3392+
/// <see cref="TreeEnsembleModelParametersBasedOnRegressionTree"/> is derived from
3393+
/// <see cref="TreeEnsembleModelParameters"/> plus a strongly-typed public attribute,
3394+
/// <see cref="TrainedTreeEnsemble"/>, for exposing trained model's details to users.
3395+
/// Its function, <see cref="CreateTreeEnsembleFromInternalDataStructure"/>, is
3396+
/// called to create <see cref="TrainedTreeEnsemble"/> inside <see cref="TreeEnsembleModelParameters"/>.
3397+
/// Note that the major difference between <see cref="TreeEnsembleModelParametersBasedOnQuantileRegressionTree"/>
3398+
/// and <see cref="TreeEnsembleModelParametersBasedOnRegressionTree"/> is the type of
3399+
/// <see cref="TrainedTreeEnsemble"/>.
3400+
/// </summary>
33853401
public abstract class TreeEnsembleModelParametersBasedOnRegressionTree : TreeEnsembleModelParameters
33863402
{
33873403
private TreeEnsemble<RegressionTree> _trainedTreeEnsemble;
@@ -3403,6 +3419,9 @@ protected TreeEnsembleModelParametersBasedOnRegressionTree(IHostEnvironment env,
34033419
{
34043420
}
34053421

3422+
/// <summary>
3423+
/// See <see cref="TreeEnsembleModelParameters.CreateTreeEnsembleFromInternalDataStructure"/>.
3424+
/// </summary>
34063425
protected override void CreateTreeEnsembleFromInternalDataStructure()
34073426
{
34083427
var trees = TrainedEnsemble.Trees.Select(tree => new RegressionTree(tree));
@@ -3411,6 +3430,16 @@ protected override void CreateTreeEnsembleFromInternalDataStructure()
34113430
}
34123431
}
34133432

3433+
/// <summary>
3434+
/// <see cref="TreeEnsembleModelParametersBasedOnQuantileRegressionTree"/> is derived from
3435+
/// <see cref="TreeEnsembleModelParameters"/> plus a strongly-typed public attribute,
3436+
/// <see cref="TrainedTreeEnsemble"/>, for exposing trained model's details to users.
3437+
/// Its function, <see cref="CreateTreeEnsembleFromInternalDataStructure"/>, is
3438+
/// called to create <see cref="TrainedTreeEnsemble"/> inside <see cref="TreeEnsembleModelParameters"/>.
3439+
/// Note that the major difference between <see cref="TreeEnsembleModelParametersBasedOnQuantileRegressionTree"/>
3440+
/// and <see cref="TreeEnsembleModelParametersBasedOnRegressionTree"/> is the type of
3441+
/// <see cref="TrainedTreeEnsemble"/>.
3442+
/// </summary>
34143443
public abstract class TreeEnsembleModelParametersBasedOnQuantileRegressionTree : TreeEnsembleModelParameters
34153444
{
34163445
private TreeEnsemble<QuantileRegressionTree> _trainedTreeEnsemble;
@@ -3432,6 +3461,9 @@ protected TreeEnsembleModelParametersBasedOnQuantileRegressionTree(IHostEnvironm
34323461
{
34333462
}
34343463

3464+
/// <summary>
3465+
/// See <see cref="TreeEnsembleModelParameters.CreateTreeEnsembleFromInternalDataStructure"/>.
3466+
/// </summary>
34353467
protected override void CreateTreeEnsembleFromInternalDataStructure()
34363468
{
34373469
var trees = TrainedEnsemble.Trees.Select(tree => new QuantileRegressionTree((InternalQuantileRegressionTree)tree));

0 commit comments

Comments
 (0)