Skip to content

Commit 1b4474c

Browse files
committed
Address comments
1 parent 004d4cc commit 1b4474c

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

src/Microsoft.ML.FastTree/FastTreeArguments.cs

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@ internal override void Check(IExceptionContext ectx)
273273
#if OLD_DATALOAD
274274
ectx.CheckUserArg(0 <= secondaryMetricShare && secondaryMetricShare <= 1, "secondaryMetricShare", "secondaryMetricShare must be between 0 and 1.");
275275
#endif
276-
ectx.CheckUserArg(0 < NdcgTruncationLevel, nameof(NdcgTruncationLevel), "lambdaMartMaxTruncation must be positive.");
276+
ectx.CheckUserArg(0 < NdcgTruncationLevel, nameof(NdcgTruncationLevel), "must be positive.");
277277
}
278278
}
279279
}
@@ -628,9 +628,10 @@ public enum OptimizationAlgorithmType { GradientDescent, AcceleratedGradientDesc
628628
/// <summary>
629629
/// Early stopping metrics. (For regression, 1: L1, 2:L2; for ranking, 1:NDCG@1, 3:NDCG@3).
630630
/// </summary>
631+
[BestFriend]
631632
[Argument(ArgumentType.AtMostOnce, HelpText = "Early stopping metrics. (For regression, 1: L1, 2:L2; for ranking, 1:NDCG@1, 3:NDCG@3)", ShortName = "esmt")]
632633
[TGUI(Description = "Early stopping metrics. (For regression, 1: L1, 2:L2; for ranking, 1:NDCG@1, 3:NDCG@3)")]
633-
public int EarlyStoppingMetrics;
634+
internal int EarlyStoppingMetrics;
634635

635636
/// <summary>
636637
/// Enable post-training pruning to avoid overfitting. (a validation set is required).
@@ -732,24 +733,27 @@ public enum OptimizationAlgorithmType { GradientDescent, AcceleratedGradientDesc
732733
/// <summary>
733734
/// Freeform defining the scores that should be used as the baseline ranker.
734735
/// </summary>
736+
[BestFriend]
735737
[Argument(ArgumentType.LastOccurenceWins, HelpText = "Freeform defining the scores that should be used as the baseline ranker", ShortName = "basescores", Hide = true)]
736738
[TGUI(NotGui = true)]
737-
public string BaselineScoresFormula;
739+
internal string BaselineScoresFormula;
738740

739741
/// <summary>
740742
/// Baseline alpha for tradeoffs of risk (0 is normal training).
741743
/// </summary>
744+
[BestFriend]
742745
[Argument(ArgumentType.LastOccurenceWins, HelpText = "Baseline alpha for tradeoffs of risk (0 is normal training)", ShortName = "basealpha", Hide = true)]
743746
[TGUI(NotGui = true)]
744-
public string BaselineAlphaRisk;
747+
internal string BaselineAlphaRisk;
745748

746749
/// <summary>
747750
/// The discount freeform which specifies the per position discounts of examples in a query (uses a single variable P for position where P=0 is first position).
748751
/// </summary>
752+
[BestFriend]
749753
[Argument(ArgumentType.LastOccurenceWins, HelpText = "The discount freeform which specifies the per position discounts of examples in a query (uses a single variable P for position where P=0 is first position)",
750754
ShortName = "pdff", Hide = true)]
751755
[TGUI(NotGui = true)]
752-
public string PositionDiscountFreeform;
756+
internal string PositionDiscountFreeform;
753757

754758
#if !NO_STORE
755759
[Argument(ArgumentType.LastOccurenceWins, HelpText = "Offload feature bins to a file store", ShortName = "fbsopt", Hide = true)]

0 commit comments

Comments
 (0)