diff --git a/src/Microsoft.ML.FastTree/FastTreeRanking.cs b/src/Microsoft.ML.FastTree/FastTreeRanking.cs index 7f89858e008..cf17a0d2bee 100644 --- a/src/Microsoft.ML.FastTree/FastTreeRanking.cs +++ b/src/Microsoft.ML.FastTree/FastTreeRanking.cs @@ -343,6 +343,9 @@ protected override void CustomizedTrainingIteration(RegressionTree tree) /// standard test for the dataset private Test CreateStandardTest(Dataset dataset) { + if (dataset.MaxDcg.Length == 0) + dataset.Skeleton.RecomputeMaxDcg(10); + return new NdcgTest( ConstructScoreTracker(dataset), dataset.Ratings,