Skip to content

Commit ff94c5d

Browse files
fix: fix calculating power law r2
1 parent e84d5b7 commit ff94c5d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graphgen/models/evaluator/kg/structure_evaluator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,8 +97,8 @@ def _calculate_powerlaw_r2(degree_map: Dict[str, int]) -> Optional[float]:
9797
log_frequencies = np.log(frequencies)
9898

9999
# Linear regression on log-log scale
100-
r_value, *_ = stats.linregress(log_degrees, log_frequencies)
101-
r2 = r_value**2
100+
result = stats.linregress(log_degrees, log_frequencies)
101+
r2 = result.rvalue**2
102102

103103
return float(r2)
104104
except Exception as e:

0 commit comments

Comments
 (0)