mirror of
https://github.com/jihe520/MathModelAgent.git
synced 2026-09-19 08:09:48 +08:00
Sanjin
78 lines
3.3 KiB
Python
78 lines
3.3 KiB
Python
import os
|
||
from core.LLM import DeepSeekModel
|
||
from utils.cli import UserInput
|
||
from utils.data_recorder import DataRecorder
|
||
from utils.logger import log
|
||
from utils.common_utils import create_work_directories, create_task_id, load_toml
|
||
from config.config import Config
|
||
from models.task import Task
|
||
from models.user_output import UserOutput
|
||
|
||
|
||
def start():
|
||
log.set_console_level("WARNING")
|
||
|
||
# 初始化日志系统，设置日志目录
|
||
task_id = create_task_id()
|
||
base_dir, dirs = create_work_directories(task_id)
|
||
log.init(dirs["log"])
|
||
|
||
config = Config(load_toml(os.path.join("config", "config.toml")))
|
||
|
||
data_recorder = DataRecorder(dirs["log"])
|
||
|
||
# 加载配置文件
|
||
deepseek_model = DeepSeekModel(
|
||
**config.get_model_config(), data_recorder=data_recorder
|
||
)
|
||
|
||
# 直接设置questions示例
|
||
test_questions = {
|
||
"title": "农作物种植优化问题",
|
||
"background": """ 母亲是婴儿生命中最重要的人之一，她不仅为婴儿提供营养物质和身体保护，
|
||
还为婴儿提供情感支持和安全感。母亲心理健康状态的不良状况，如抑郁、焦虑、
|
||
压力等，可能会对婴儿的认知、情感、社会行为等方面产生负面影响。压力过大的
|
||
母亲的可能会对婴儿的生理和心理发展产生负面影响，例如影响其睡眠等方面。
|
||
附件给出了包括 390名 3 至 12 个月婴儿以及其母亲的相关数据。这些数
|
||
据涵盖各种主题，母亲的身体指标包括年龄、婚姻状况、教育程度、妊娠时间、
|
||
分娩方式，以及产妇心理指标CBTS（分娩相关创伤后应激障碍问卷）、EPDS
|
||
（爱丁堡产后抑郁量表）、HADS（医院焦虑抑郁量表）和婴儿睡眠质量指标包
|
||
括整晚睡眠时间、睡醒次数和入睡方式。
|
||
请查阅相关文献，了解专业背景，根据题目数据建立数学模型，回答下列问
|
||
题。""",
|
||
"ques_count": 2,
|
||
"ques1": """ 1. 许多研究表明，母亲的身体指标和心理指标对婴儿的行为特征和睡眠质
|
||
量有影响，请问是否存在这样的规律，根据附件中的数据对此进行研究。""",
|
||
"ques2": """ 2. 婴儿行为问卷是一个用于评估婴儿行为特征的量表，其中包含了一些关
|
||
于婴儿情绪和反应的问题。我们将婴儿的行为特征分为三种类型：安静型、中等
|
||
型、矛盾型。请你建立婴儿的行为特征与母亲的身体指标与心理指标的关系模型。
|
||
数据表中最后有20组（编号391-410号）婴儿的行为特征信息被删除，请你判断
|
||
他们是属于什么类型。""",
|
||
}
|
||
|
||
user_input = UserInput(
|
||
data_folder_path="./project/sample_data",
|
||
bg_ques_all="原始问题文本...",
|
||
model=deepseek_model,
|
||
init_with_llm=False, # 不使用LLM初始化
|
||
)
|
||
|
||
user_input.set_config_template(config.get_config_template(user_input.comp_template))
|
||
user_input.set_questions_directly(test_questions)
|
||
|
||
task = Task(
|
||
task_id=task_id,
|
||
base_dir=base_dir,
|
||
work_dirs=dirs,
|
||
llm=deepseek_model,
|
||
config=config,
|
||
)
|
||
|
||
user_output: UserOutput = task.run(user_input, data_recorder=data_recorder)
|
||
|
||
user_output.save_result(ques_count=user_input.get_ques_count())
|
||
|
||
|
||
if __name__ == "__main__":
|
||
start()
|