Fix crash in Microsoft.ML.Recommender with validation set (#7196)

dotnet · Jul 24, 2024 · f51eaa2 · f51eaa2
1 parent 05bb0e3
commit f51eaa2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Microsoft.ML.Recommender/SafeTrainingAndModelBuffer.cs b/src/Microsoft.ML.Recommender/SafeTrainingAndModelBuffer.cs
@@ -368,7 +368,7 @@ public unsafe void TrainWithValidation(IChannel ch, int rowCount, int colCount,
                 validProb.R = validNodesPtrs;
                 validProb.M = rowCount;
                 validProb.N = colCount;
-                validProb.Nnz = nodes.Length;
+                validProb.Nnz = validNodes.Length;
 
                 ch.Info("Training {0} by {1} problem on {2} examples with a {3} by {4} validation set including {5} examples",
                     prob.M, prob.N, prob.Nnz, validProb.M, validProb.N, validProb.Nnz);