Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

Add jga metric to taskmaster2 #3134

Merged
merged 1 commit into from Sep 30, 2020
Merged

Add jga metric to taskmaster2 #3134

merged 1 commit into from Sep 30, 2020

Conversation

moyapchen
Copy link
Contributor

@moyapchen moyapchen commented Sep 30, 2020

Fairly straightforward.

Test Plan:

Ran an eval model with the model that I had trained previously on taskmaster2 w/ PPL as a metric locally.

Note: I'm currently running a quick local training with jga as the
metric to make sure it's functioning properly (cause trying `parlai
eval_model -m bart -t taskmaster2` off the bat has 0s in both `slot_r` and `jga` right now), but figured the metric is easy enough to warrant getting review on it sooner rather than later.
@github-actions
Copy link

Your PR contains a change to a task. Please paste the results of the following command into a comment:

python tests/datatests/test_new_tasks.py

@@ -171,6 +171,9 @@ def custom_evaluation(
f'{domain}_slot_r',
AverageMetric(correct, len(teacher_action['slots'])),
)
self.metrics.add(
'jga', AverageMetric(correct == len(teacher_action['slots']))
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It might be more readable for us to just compare the dictionaries (see FB:multiwoz code), but this works.

@moyapchen moyapchen merged commit 4aa648f into master Sep 30, 2020
@moyapchen moyapchen deleted the jga branch September 30, 2020 03:14
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants