-
Notifications
You must be signed in to change notification settings - Fork 1.9k
FastTreeRankingTrainer expose non-advanced args(#1246) #1393
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
string groupIdColumn = null, | ||
int numLeaves = Defaults.NumLeaves, | ||
int numTrees = Defaults.NumTrees, | ||
int minDocumentsInLeafs = Defaults.MinDocumentsInLeafs, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
= Defaults.MinDocumentsInLeafs [](start = 35, length = 31)
please remove defaults from the base classes
string featureColumn, | ||
string weightColumn = null, | ||
string groupIdColumn = null, | ||
int numLeaves = Defaults.NumLeaves, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
= Defaults.NumLeaves [](start = 25, length = 21)
same here; remove defaults
@@ -30,7 +30,7 @@ protected RandomForestTrainerBase(IHostEnvironment env, TArgs args, SchemaShape. | |||
/// </summary> | |||
protected RandomForestTrainerBase(IHostEnvironment env, SchemaShape.Column label, string featureColumn, | |||
string weightColumn = null, string groupIdColumn = null, bool quantileEnabled = false, Action<TArgs> advancedSettings = null) | |||
: base(env, label, featureColumn, weightColumn, groupIdColumn, advancedSettings) | |||
: base(env, label, featureColumn, weightColumn, groupIdColumn, advancedSettings: advancedSettings) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
advancedSettings: advancedSettings) [](start = 75, length = 35)
did you want to add the other args?
@@ -59,7 +59,7 @@ public sealed partial class FastTreeTweedieTrainer | |||
/// <param name="advancedSettings">A delegate to apply all the advanced arguments to the algorithm.</param> | |||
public FastTreeTweedieTrainer(IHostEnvironment env, string labelColumn, string featureColumn, | |||
string groupIdColumn = null, string weightColumn = null, Action<Arguments> advancedSettings = null) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
null, Action advancedSettings = null [](start = 63, length = 47)
would you like to add the other args
Thanks for the help @feiyun0112. |
Fix #1246
add
numLeaves
,numTrees
,minDocumentsInLeafs
,learningRate
args