Merge pull request apachecn#520 from cclauss/patch-1

Simplify ret creation with dict comprehension
wyvern92 · May 22, 2019 · ded207f · ded207f
2 parents 327f48d + cbdde1f
commit ded207f
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/py3.x/16.RecommenderSystems/test_lfm.py b/src/py3.x/16.RecommenderSystems/test_lfm.py
@@ -3,10 +3,7 @@
 
 # 负样本采样过程
 def RandomSelectNegativeSample(self, items):
-    ret = dict()
-    for i in items.keys():
-        ret[i] = 1
-
+    ret = {key: 1 for key in items}
     n = 0
     for i in range(0, len(items) * 3):
         item = items_pool[random.randint(0, len(items_pool) - 1)]