improve performance 10%

Oceania2018 · Oceania2018 · commit 2461d2f461bd · 2019-04-08T22:25:08.000-05:00
diff --git a/test/TensorFlowNET.Examples/LogisticRegression.cs b/test/TensorFlowNET.Examples/LogisticRegression.cs
@@ -53,7 +53,6 @@ public bool Run()
 
             return with(tf.Session(), sess =>
             {
-                
                 // Run the initializer
                 sess.run(init);
 
@@ -71,7 +70,7 @@ public bool Run()
                             new FeedItem(x, batch_xs),
                             new FeedItem(y, batch_ys));
 
-                        var c = (float)result[1];
+                        float c = result[1];
                         // Compute average loss
                         avg_cost += c / total_batch;
                     }
diff --git a/test/TensorFlowNET.Examples/Utility/MnistDataSet.cs b/test/TensorFlowNET.Examples/Utility/MnistDataSet.cs
@@ -96,7 +96,7 @@ private static NDArray dense_to_one_hot(NDArray labels_dense, int num_classes)
             for(int row = 0; row < num_labels; row++)
             {
                 var col = labels_dense.Data<byte>(row);
-                labels_one_hot[row, col] = 1;
+                labels_one_hot.SetData(1, row, col);
             }
 
             return labels_one_hot;

Original file line number	Diff line number	Diff line change
`@@ -96,7 +96,7 @@ private static NDArray dense_to_one_hot(NDArray labels_dense, int num_classes)`
`96`	`96`	`for(int row = 0; row < num_labels; row++)`
`97`	`97`	`{`
`98`	`98`	`var col = labels_dense.Data<byte>(row);`
`99`		`- labels_one_hot[row, col] = 1;`
	`99`	`+ labels_one_hot.SetData(1, row, col);`
`100`	`100`	`}`
`101`	`101`
`102`	`102`	`return labels_one_hot;`