search

tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning
Apache License 2.0
3.18k stars 439 forks source link

Fix predict data feeder to return non-overlapping batches #114

Closed elqursh closed 8 years ago

elqursh commented 8 years ago

setup_predict_data_feeder when given an iterator returns incorrect batches. The incorrect batches take the following form.

[0, ..., batch_size] [0, ..., batch_size, batch_size+1] ...

This commit fixes this to return non-overlapping batches of sizes equal to batch_size.

terrytangyuan commented 8 years ago

@elqursh Thanks! Will merge when tests pass.

terrytangyuan commented 8 years ago

py3 test failed, use six.next() instead.

codecov-io commented 8 years ago

Current coverage is 90.70%

Merging #114 into master will increase coverage by +0.20% as of 7407938

@@            master   #114   diff @@
=====================================
  Files           27     27       
  Stmts         1043   1044     +1
  Branches       159    159       
  Methods          0      0       
=====================================
+ Hit            944    947     +3
+ Partial         47     46     -1
+ Missed          52     51     -1

Review entire Coverage Diff as of 7407938

Powered by Codecov. Updated on successful CI builds.