diff --git a/eng/Versions.props b/eng/Versions.props
index 92c25fc465..4f025bdf04 100644
--- a/eng/Versions.props
+++ b/eng/Versions.props
@@ -38,7 +38,7 @@
     <!-- Other product dependencies -->
     <ApacheArrowVersion>14.0.2</ApacheArrowVersion>
     <GoogleProtobufVersion>3.27.1</GoogleProtobufVersion>
-    <LightGBMVersion>3.3.5</LightGBMVersion>
+    <LightGBMVersion>4.6.0</LightGBMVersion>
     <MicrosoftBclHashCodeVersion>6.0.0</MicrosoftBclHashCodeVersion>
     <MicrosoftBclMemoryVersion>9.0.0</MicrosoftBclMemoryVersion>
     <MicrosoftCodeAnalysisAnalyzersVersion>3.3.4</MicrosoftCodeAnalysisAnalyzersVersion>
diff --git a/src/Microsoft.ML.LightGbm/WrappedLightGbmDataset.cs b/src/Microsoft.ML.LightGbm/WrappedLightGbmDataset.cs
index 4f9ae19f3d..713dfa8159 100644
--- a/src/Microsoft.ML.LightGbm/WrappedLightGbmDataset.cs
+++ b/src/Microsoft.ML.LightGbm/WrappedLightGbmDataset.cs
@@ -67,7 +67,7 @@ public unsafe Dataset(double[][] sampleValuePerColumn,
                 {
                     // Create container. Examples will pushed in later.
                     LightGbmInterfaceUtils.Check(WrappedLightGbmInterface.DatasetCreateFromSampledColumn(
-                        (IntPtr)ptrValues, (IntPtr)ptrIndices, numCol, sampleNonZeroCntPerColumn, numSampleRow, numTotalRow,
+                        (IntPtr)ptrValues, (IntPtr)ptrIndices, numCol, sampleNonZeroCntPerColumn, numSampleRow, numTotalRow, 0,
                         param, out _handle));
                 }
             }
diff --git a/src/Microsoft.ML.LightGbm/WrappedLightGbmInterface.cs b/src/Microsoft.ML.LightGbm/WrappedLightGbmInterface.cs
index 7cbcea76a1..acb7ad481c 100644
--- a/src/Microsoft.ML.LightGbm/WrappedLightGbmInterface.cs
+++ b/src/Microsoft.ML.LightGbm/WrappedLightGbmInterface.cs
@@ -87,7 +87,8 @@ public static extern int DatasetCreateFromSampledColumn(IntPtr sampleValuePerCol
             int numCol,
             int[] sampleNonZeroCntPerColumn,
             int numSampleRow,
-            int numTotalRow,
+            int numTotalLocalRow,
+            int numTotalDistributedRow,
             [MarshalAs(UnmanagedType.LPStr)] string parameters,
             out SafeDataSetHandle ret);
 
diff --git a/test/Microsoft.ML.TensorFlow.Tests/TensorflowTests.cs b/test/Microsoft.ML.TensorFlow.Tests/TensorflowTests.cs
index 665be60361..16bc4a6b74 100644
--- a/test/Microsoft.ML.TensorFlow.Tests/TensorflowTests.cs
+++ b/test/Microsoft.ML.TensorFlow.Tests/TensorflowTests.cs
@@ -675,7 +675,7 @@ public void TensorFlowTransformMNISTConvTest()
             var metrics = _mlContext.MulticlassClassification.Evaluate(predicted);
 
             Assert.Equal(0.99, metrics.MicroAccuracy, .01);
-            Assert.Equal(0.93, metrics.MacroAccuracy, 0.01);
+            Assert.Equal(0.99, metrics.MacroAccuracy, 0.01);
 
             var oneSample = GetOneMNISTExample();
 
@@ -900,7 +900,7 @@ public void TensorFlowTransformMNISTConvSavedModelTest()
 
             // First group of checks
             Assert.Equal(0.99, metrics.MicroAccuracy, .01);
-            Assert.Equal(.93, metrics.MacroAccuracy, 0.01);
+            Assert.Equal(.99, metrics.MacroAccuracy, 0.01);
 
             // An in-memory example. Its label is predicted below.
             var oneSample = GetOneMNISTExample();