Skip to content

Commit 1ddc9b6

Browse files
committed
run LinearRegression but failed.
docs updated.
1 parent 8408dcd commit 1ddc9b6

3 files changed

Lines changed: 27 additions & 4 deletions

File tree

docs/source/Train.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
# Chapter. Trainer
2+
3+
### Saver
4+
5+
The `tf.train.saver` class provides methods to save and restore models.

docs/source/index.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,4 +26,5 @@ Welcome to TensorFlow.NET's documentation!
2626
NameScope
2727
ControlDependency
2828
Gradient
29+
Train
2930
EagerMode

test/TensorFlowNET.Examples/LinearRegression.cs

Lines changed: 21 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,15 +61,32 @@ public void Run()
6161
sess.run(init);
6262

6363
// Fit all training data
64-
for (int i = 0; i < training_epochs; i++)
64+
for (int epoch = 0; epoch < training_epochs; epoch++)
6565
{
66-
foreach(var (x, y) in Python.zip<double>(train_X, train_Y))
66+
foreach (var (x, y) in Python.zip<double>(train_X, train_Y))
6767
{
68-
var feed_dict = new Dictionary<Tensor, NDArray>();
68+
sess.run(optimizer, feed_dict: new FeedItem[]
69+
{
70+
new FeedItem(X, x),
71+
new FeedItem(Y, y)
72+
});
73+
}
74+
75+
// Display logs per epoch step
76+
if ((epoch + 1) % display_step == 0)
77+
{
78+
var c = sess.run(cost, feed_dict: new FeedItem[]
79+
{
80+
new FeedItem(X, train_X),
81+
new FeedItem(Y, train_Y)
82+
});
6983

70-
// sess.run(optimizer, feed_dict);
84+
Console.WriteLine($"Epoch: {epoch + 1} cost={c} " +
85+
$"W={sess.run(W)} b={sess.run(b)}");
7186
}
7287
}
88+
89+
Console.WriteLine("Optimization Finished!");
7390
});
7491
}
7592
}

0 commit comments

Comments
 (0)