File size: 2,397 Bytes
0c51b93 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
from sotopia_rl.prompter.all_the_same_attribution_function import (
get_attribution_single_conv as all_the_same_attribution_normalized_single_conv,
)
from sotopia_rl.prompter.direct_attribution_function import (
get_attribution_single_conv as direct_attribution_single_conv,
)
from sotopia_rl.prompter.direct_attribution_generic_function import (
get_attribution_single_conv as direct_attribution_generic_single_conv,
)
from sotopia_rl.prompter.direct_attribution_normalized_function import (
get_attribution_single_conv as direct_attribution_normalized_single_conv,
)
from sotopia_rl.prompter.discounting_attribution_function import (
get_attribution_single_conv as discounting_attribution_single_conv,
)
from sotopia_rl.prompter.goal_progress_attribution_function import (
get_attribution_single_conv as goal_progress_attribution_single_conv,
)
from sotopia_rl.prompter.key_utterance_function import (
get_attribution_single_conv as key_utterance_attribution_single_conv,
)
from sotopia_rl.prompter.only_response_attribution_function import (
get_attribution_single_conv as only_response_attribution_single_conv,
)
from sotopia_rl.prompter.utterance_quality_attribution_function import (
get_attribution_single_conv as utterance_quality_attribution_single_conv,
)
from sotopia_rl.prompter.utterance_quality_attribution_normalized_function import (
get_attribution_single_conv as utterance_quality_attribution_normalized_single_conv,
)
# from sotopia_rl.prompter.utterance_quality_generic_function import (
# get_attribution_single_conv as utterance_quality_attribution_generic_single_conv,
# )
ATTRIBUTION_METHOD_DICT = {
"discounting": discounting_attribution_single_conv,
"direct": direct_attribution_single_conv,
"direct_normalized": direct_attribution_normalized_single_conv,
"goal_progress": goal_progress_attribution_single_conv,
"only_response": only_response_attribution_single_conv,
"utterance_quality": utterance_quality_attribution_single_conv,
"utterance_quality_normalized": utterance_quality_attribution_normalized_single_conv,
"all_the_same": all_the_same_attribution_normalized_single_conv,
"key_utterance": key_utterance_attribution_single_conv,
"direct_generic": direct_attribution_generic_single_conv,
# "utterance_quality_generic": utterance_quality_attribution_generic_single_conv,
}
|