使用场景

研究人员使用ExploreToM生成的数据训练心理理论推理模型。

教育机构利用该框架创建教学案例，帮助学生理解心理理论。

开发者使用ExploreToM框架来测试和改进他们的聊天机器人或虚拟助手。

产品特色

生成故事背景：使用story_context_generator.py脚本生成故事背景。

执行A*搜索：通过story_structure_searcher.py脚本执行A*搜索，以产生复杂的故事结构。

填充生成的故事：利用story_structure_infiller.py脚本填充生成的故事。

统计分析：通过compute_statistics.py脚本对生成的数据进行统计分析。

功能测试：运行tests_belief_tracker.py和tests_story_structure_infiller.py进行功能测试。

模型加载：使用VLLM（大型语言模型）加载和运行模型。

使用教程

1. 安装必要的Python环境和依赖。

2. 使用story_context_generator.py生成故事背景。

3. 通过story_structure_searcher.py执行A*搜索，生成复杂的故事结构。

4. 利用story_structure_infiller.py填充生成的故事。

5. 运行compute_statistics.py对生成的数据进行统计分析。

6. 通过tests_belief_tracker.py和tests_story_structure_infiller.py进行功能测试。

7. 根据需要加载和使用VLLM模型。