Zekun Wu commited on
Commit
286c449
1 Parent(s): 0356c02
Files changed (1) hide show
  1. util/evaluation.py +2 -0
util/evaluation.py CHANGED
@@ -20,6 +20,7 @@ def statistical_tests(data):
20
  # Calculate average ranks
21
  rank_columns = [v + rank_suffix for v in variables]
22
  average_ranks = data[rank_columns].mean()
 
23
 
24
  # Statistical tests
25
  rank_data = [data[col] for col in rank_columns]
@@ -56,6 +57,7 @@ def statistical_tests(data):
56
 
57
  results = {
58
  "Average Ranks": average_ranks.to_dict(),
 
59
  "Friedman Test": {
60
  "Statistic": friedman_stat,
61
  "p-value": friedman_p,
 
20
  # Calculate average ranks
21
  rank_columns = [v + rank_suffix for v in variables]
22
  average_ranks = data[rank_columns].mean()
23
+ average_scores = data[[v + score_suffix for v in variables]].mean()
24
 
25
  # Statistical tests
26
  rank_data = [data[col] for col in rank_columns]
 
57
 
58
  results = {
59
  "Average Ranks": average_ranks.to_dict(),
60
+ "Average Scores": average_scores.to_dict(),
61
  "Friedman Test": {
62
  "Statistic": friedman_stat,
63
  "p-value": friedman_p,