small changes

etlundquist · Jun 8, 2020 · 82720e6 · 82720e6
1 parent 285baf1
commit 82720e6
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git a/Makefile b/Makefile
@@ -1,5 +1,5 @@
 lint:
- python -m pyflakes rankfm/
+ python -m flake8 --ignore W3,E3,E5,E74 rankfm/
 
 test:
  python -m pytest -r Efp tests/
diff --git a/rankfm/numba_methods.py b/rankfm/numba_methods.py
@@ -102,7 +102,7 @@ def _fit(interactions, sample_weight, user_items, item_idx, regularization, lear
 
  # create a shuffle index to diversify each training epoch
  n_interaction = len(interactions)
- shuffle_index = np.arange(n_interaction)
+ shuffle_index = np.arange(n_interaction, dtype=np.int32)
 
  # start the main training epoch loop
 

diff --git a/rankfm/rankfm.py b/rankfm/rankfm.py
@@ -9,10 +9,6 @@
 from rankfm.numba_methods import _fit, _predict, _recommend
 from rankfm.utils import get_data
 
-# import warnings
-# warnings.filterwarnings("ignore", category=nb.NumbaPerformanceWarning)
-
-
 class RankFM():
  """Factorization Machines for Ranking Problems with Implicit Feedback Data"""