A bug: num_batches_per_epoch = int(data_size / batch_size) + 1

jiegzhan / multi-class-text-classification-cnn

Classify Kaggle Consumer Finance Complaints into 11 classes. Build the model with CNN (Convolutional Neural Network) and Word Embeddings on Tensorflow.

Apache License 2.0

426 stars 198 forks source link

A bug: num_batches_per_epoch = int(data_size / batch_size) + 1 #21

Open lemo2012 opened 6 years ago

lemo2012 commented 6 years ago

should be: num_batches_per_epoch = int(math.ceil(float(data_size) / batch_size))