@@ -27,7 +27,7 @@ public partial class Model
2727 /// <param name="use_multiprocessing"></param>
2828 /// <param name="return_dict"></param>
2929 /// <param name="is_val"></param>
30- public Dictionary < string , float > evaluate ( NDArray x , NDArray y ,
30+ public Dictionary < string , float > evaluate ( Tensor x , Tensor y ,
3131 int batch_size = - 1 ,
3232 int verbose = 1 ,
3333 int steps = - 1 ,
@@ -91,7 +91,7 @@ public Dictionary<string, float> evaluate(NDArray x, NDArray y,
9191 return results ;
9292 }
9393
94- public Dictionary < string , float > evaluate ( IEnumerable < Tensor > x , NDArray y , int verbose = 1 , bool is_val = false )
94+ public Dictionary < string , float > evaluate ( IEnumerable < Tensor > x , Tensor y , int verbose = 1 , bool is_val = false )
9595 {
9696 var data_handler = new DataHandler ( new DataHandlerArgs
9797 {
@@ -119,7 +119,7 @@ public Dictionary<string, float> evaluate(IEnumerable<Tensor> x, NDArray y, int
119119 foreach ( var step in data_handler . steps ( ) )
120120 {
121121 callbacks . on_test_batch_begin ( step ) ;
122- logs = test_step_multi_inputs_function ( data_handler , iterator ) ;
122+ logs = test_function ( data_handler , iterator ) ;
123123 var end_step = step + data_handler . StepIncrement ;
124124 if ( is_val == false )
125125 callbacks . on_test_batch_end ( end_step , logs ) ;
@@ -178,20 +178,14 @@ public Dictionary<string, float> evaluate(IDatasetV2 x, int verbose = 1, bool is
178178 }
179179
180180 Dictionary < string , float > test_function ( DataHandler data_handler , OwnedIterator iterator )
181- {
182- var data = iterator . next ( ) ;
183- var outputs = test_step ( data_handler , data [ 0 ] , data [ 1 ] ) ;
184- tf_with ( ops . control_dependencies ( new object [ 0 ] ) , ctl => _test_counter . assign_add ( 1 ) ) ;
185- return outputs ;
186- }
187- Dictionary < string , float > test_step_multi_inputs_function ( DataHandler data_handler , OwnedIterator iterator )
188181 {
189182 var data = iterator . next ( ) ;
190183 var x_size = data_handler . DataAdapter . GetDataset ( ) . FirstInputTensorCount ;
191184 var outputs = train_step ( data_handler , new Tensors ( data . Take ( x_size ) ) , new Tensors ( data . Skip ( x_size ) ) ) ;
192- tf_with ( ops . control_dependencies ( new object [ 0 ] ) , ctl => _train_counter . assign_add ( 1 ) ) ;
185+ tf_with ( ops . control_dependencies ( new object [ 0 ] ) , ctl => _test_counter . assign_add ( 1 ) ) ;
193186 return outputs ;
194187 }
188+
195189 Dictionary < string , float > test_step ( DataHandler data_handler , Tensor x , Tensor y )
196190 {
197191 ( x , y ) = data_handler . DataAdapter . Expand1d ( x , y ) ;
0 commit comments