We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 19dc713 commit 280bda1Copy full SHA for 280bda1
trinity/utils/monitor.py
@@ -69,7 +69,7 @@ def gather_eval_metrics(
69
for col in numeric_df.columns:
70
# Skip the columns that are already aggregated
71
key_words = ["std", "mean", "min", "max"]
72
- if any(key_word in col.lower() for key_word in key_words):
+ if any(col.endswith(key_word) for key_word in key_words):
73
metric[f"{prefix}/{col}"] = numeric_df[col].mean()
74
else:
75
stats_df = numeric_df[[col]].agg(output_stats)
0 commit comments