Bases: LMRewardsOperationalMetric
Input (prompt) tokens consumed by LM calls during reward computation.
Example:
program.compile(
metrics=[
synalinks.metrics.RewardInputTokens(),
],
)
Source code in synalinks/src/metrics/lm_metrics.py
| @synalinks_export("synalinks.metrics.RewardInputTokens")
class RewardInputTokens(LMRewardsOperationalMetric):
"""Input (prompt) tokens consumed by LM calls during reward computation.
Example:
```python
program.compile(
metrics=[
synalinks.metrics.RewardInputTokens(),
],
)
```
"""
def __init__(self, name="reward_input_tokens"):
super().__init__(name=name)
def result(self):
return int(self._delta("prompt_tokens"))
|