From c413b32c9e18cd84232dabad0261622eb2f0d59f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=A8=E5=A9=89?= <yangwan@yangwandeMacBook-Pro.local>
Date: Sun, 21 Apr 2019 22:18:50 +0800
Subject: [PATCH] fit generator bug

---
 dynamicgem/embedding/sdne_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dynamicgem/embedding/sdne_utils.py b/dynamicgem/embedding/sdne_utils.py
index dd43c8e..5704c8f 100644
--- a/dynamicgem/embedding/sdne_utils.py
+++ b/dynamicgem/embedding/sdne_utils.py
@@ -57,7 +57,7 @@ def model_batch_predictor_v2(model, X, batch_size):
         pdb.set_trace()    
 
 def batch_generator_ae(X, beta, batch_size, shuffle):
-    number_of_batches = X.shape[0] / batch_size
+    number_of_batches = X.shape[0] // batch_size
     counter = 0
     sample_index = np.arange(X.shape[0])
     if shuffle: