Skip to content

Commit 6514bd2

Browse files
committed
PR feedback.
1 parent 610bdb6 commit 6514bd2

File tree

2 files changed

+7
-26
lines changed

2 files changed

+7
-26
lines changed

src/Microsoft.ML.FastTree/FastTreeRanking.cs

+4-23
Original file line numberDiff line numberDiff line change
@@ -52,35 +52,16 @@ namespace Microsoft.ML.Trainers.FastTree
5252
/// | | |
5353
/// | -- | -- |
5454
/// | Machine learning task | Ranking |
55-
/// | Is normalization required? | Yes |
55+
/// | Is normalization required? | No |
5656
/// | Is caching required? | No |
5757
/// | Required NuGet in addition to Microsoft.ML | Microsoft.ML.FastTree |
5858
///
59-
/// ### Training Algorithm Details
60-
/// FastTree is an efficient implementation of the [MART](https://arxiv.org/abs/1505.01866) gradient boosting algorithm.
61-
/// Gradient boosting is a machine learning technique for regression problems.
62-
/// It builds each regression tree in a step-wise fashion, using a predefined loss function to measure the error for each step and corrects for it in the next.
63-
/// So this prediction model is actually an ensemble of weaker prediction models.In regression problems, boosting builds a series of such trees in a step-wise fashion and then selects the optimal tree using an arbitrary differentiable loss function.
64-
///
65-
/// MART learns an ensemble of regression trees, which is a decision tree with scalar values in its leaves.
66-
/// A decision(or regression) tree is a binary tree-like flow chart, where at each interior node one decides which of the two child nodes to continue to based on one of the feature values from the input.
67-
/// At each leaf node, a value is returned.In the interior nodes, the decision is based on the test 'x <= v' where x is the value of the feature in the input sample and v is one of the possible values of this feature.
68-
/// The functions that can be produced by a regression tree are all the piece-wise constant functions.
69-
///
70-
/// The ensemble of trees is produced by computing, in each step, a regression tree that approximates the gradient of the loss function, and adding it to the previous tree with coefficients that minimize the loss of the new tree.
71-
/// The output of the ensemble produced by MART on a given instance is the sum of the tree outputs.
72-
/// In case of a binary classification problem, the output is converted to a probability by using some form of calibration.
73-
/// In case of a regression problem, the output is the predicted value of the function.
74-
/// In case of a ranking problem, the instances are ordered by the output value of the ensemble.
75-
///
76-
/// For more information see:
77-
/// [Wikipedia: Gradient boosting (Gradient tree boosting)](https://en.wikipedia.org/wiki/Gradient_boosting#Gradient_tree_boosting).
78-
/// [Greedy function approximation: A gradient boosting machine.](https://projecteuclid.org/DPubS?service=UI&version=1.0&verb=Display&handle=euclid.aos/1013203451)
59+
/// [!include[algorithm](~/../docs/samples/docs/api-reference/algo-details-fasttree.md)]
7960
/// ]]>
8061
/// </format>
8162
/// </remarks>
82-
/// <seealso cref="Microsoft.ML.TreeExtensions.FastTree(Microsoft.ML.RankingCatalog.RankingTrainers,System.String,System.String,System.String,System.String,System.Int32,System.Int32,System.Int32,System.Double)"/>
83-
/// <seealso cref="Microsoft.ML.TreeExtensions.FastTree(Microsoft.ML.RankingCatalog.RankingTrainers,Microsoft.ML.Trainers.FastTree.FastTreeRankingTrainer.Options)"/>
63+
/// <seealso cref="TreeExtensions.FastTree(RankingCatalog.RankingTrainers, string, string, string, string, int, int, int, double)"/>
64+
/// <seealso cref="TreeExtensions.FastTree(RegressionCatalog.RegressionTrainers, FastTreeRegressionTrainer.Options)"/>
8465
/// <seealso cref="Options"/>
8566
public sealed partial class FastTreeRankingTrainer
8667
: BoostingFastTreeTrainerBase<FastTreeRankingTrainer.Options, RankingPredictionTransformer<FastTreeRankingModelParameters>, FastTreeRankingModelParameters>

src/Microsoft.ML.FastTree/TreeTrainersCatalog.cs

+3-3
Original file line numberDiff line numberDiff line change
@@ -122,11 +122,11 @@ public static FastTreeBinaryTrainer FastTree(this BinaryClassificationCatalog.Bi
122122
}
123123

124124
/// <summary>
125-
/// Creates a <see cref="FastTreeRankingTrainer"/>, which ranks a series of inputs based on their relevance and training a decision tree ranking model.
125+
/// Create a <see cref="FastTreeRankingTrainer"/>, which ranks a series of inputs based on their relevance and training a decision tree ranking model.
126126
/// </summary>
127127
/// <param name="catalog">The <see cref="RankingCatalog"/>.</param>
128128
/// <param name="labelColumnName">The name of the label column. The column data must be <see cref="System.Single"/>.</param>
129-
/// <param name="featureColumnName">The name of the feature column. The column data must be a known-sized vector of <see cref="System.Single"/></param>
129+
/// <param name="featureColumnName">The name of the feature column. The column data must be a known-sized vector of <see cref="System.Single"/>.</param>
130130
/// <param name="rowGroupColumnName">The name of the group column.</param>
131131
/// <param name="exampleWeightColumnName">The name of the example weight column (optional).</param>
132132
/// <param name="numberOfTrees">Total number of decision trees to create in the ensemble.</param>
@@ -156,7 +156,7 @@ public static FastTreeRankingTrainer FastTree(this RankingCatalog.RankingTrainer
156156
}
157157

158158
/// <summary>
159-
/// Creates a <see cref="FastTreeRankingTrainer"/>, which ranks a series of inputs based on their relevance and training a decision tree ranking model with advanced options.
159+
/// Create a <see cref="FastTreeRankingTrainer"/> with advanced options, which ranks a series of inputs based on their relevance and training a decision tree ranking model.
160160
/// </summary>
161161
/// <param name="catalog">The <see cref="RankingCatalog"/>.</param>
162162
/// <param name="options">Trainer options.</param>

0 commit comments

Comments
 (0)