Skip to content

Commit 48e11c6

Browse files
committed
simplify StructureEvaluator
1 parent 6870dc4 commit 48e11c6

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

graphgen/models/evaluator/kg/structure_evaluator.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -83,12 +83,7 @@ def _calculate_powerlaw_r2(degree_map: Dict[str, int]) -> Optional[float]:
8383

8484
try:
8585
degree_counts = Counter(degrees)
86-
degree_values = []
87-
frequencies = []
88-
for deg, freq in sorted(degree_counts.items()):
89-
if deg > 0 and freq > 0:
90-
degree_values.append(deg)
91-
frequencies.append(freq)
86+
degree_values, frequencies = zip(*sorted(degree_counts.items()))
9287

9388
if len(degree_values) < 3:
9489
logger.warning(

0 commit comments

Comments
 (0)