File tree Expand file tree Collapse file tree
test/TensorFlowNET.Examples Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1+ # Chapter. Trainer
2+
3+ ### Saver
4+
5+ The ` tf.train.saver ` class provides methods to save and restore models.
Original file line number Diff line number Diff line change @@ -26,4 +26,5 @@ Welcome to TensorFlow.NET's documentation!
2626 NameScope
2727 ControlDependency
2828 Gradient
29+ Train
2930 EagerMode
Original file line number Diff line number Diff 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 }
You can’t perform that action at this time.
0 commit comments