Bases: LMRewardsOperationalMetric
Average total tokens per LM call during reward computation.
Example:
program.compile(
metrics=[
synalinks.metrics.AvgRewardTotalTokensPerCall(),
],
)
Source code in synalinks/src/metrics/lm_metrics.py
| @synalinks_export("synalinks.metrics.AvgRewardTotalTokensPerCall")
class AvgRewardTotalTokensPerCall(LMRewardsOperationalMetric):
"""Average total tokens per LM call during reward computation.
Example:
```python
program.compile(
metrics=[
synalinks.metrics.AvgRewardTotalTokensPerCall(),
],
)
```
"""
def __init__(self, name="avg_reward_total_tokens_per_call"):
super().__init__(name=name)
def result(self):
calls = self._delta("calls")
if calls <= 0:
return 0.0
return self._delta("tokens") / calls
|