Fixed a bug caused by a new version of NumPy. See mnick#3 for fix and m…

…nick/holographic-embeddings#4 for a reference to this issue.
eriq-augustine · Jul 12, 2017 · 28a8462 · 28a8462
1 parent fb4ccd0
commit 28a8462
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/skge/base.py b/skge/base.py
@@ -127,7 +127,7 @@ def _pre_epoch(self):
 
     def _optim(self, xys):
         idx = np.arange(len(xys))
-        self.batch_size = np.ceil(len(xys) / self.nbatches)
+        self.batch_size = int(np.ceil(len(xys) / self.nbatches))
         batch_idx = np.arange(self.batch_size, len(xys), self.batch_size)
 
         for self.epoch in range(1, self.max_epochs + 1):