Update c2pf_example

Aznoryusof · Mar 22, 2019 · 51d7cf2 · 51d7cf2
1 parent f70e37c
commit 51d7cf2
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/examples/c2pf_example.py b/examples/c2pf_example.py
@@ -23,7 +23,7 @@
                          shuffle=True, exclude_unknowns=True,
                          verbose=True, item_graph=item_graph_module)
 
-c2pf = C2PF(k=100, max_iter=0, variant='c2pf')
+c2pf = C2PF(k=100, max_iter=1, variant='c2pf')
 
 # Evaluation metrics
 nDgc = metrics.NDCG(k=-1)
@@ -36,11 +36,3 @@
                  models=[c2pf],
                  metrics=[nDgc, mrr, rec, pre])
 exp.run()
-
-import unittest
-
-            unittest.TestCase.assertEqual(self,len(ratings), 53282)
-            self.assertEqual(len(contexts), 108466)
-
-
-item_graph_module.build(global_id_map = None)