leo/khqp_trainer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

完成v1, 整理好所有参数

Browse Source
This commit is contained in:
leo
2024-03-07 23:15:38 +08:00
parent f22d134fb4
commit 7128d489da
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@@ -43,6 +43,7 @@ def make_train_dicts(with_entities: tuple, conditions: tuple):
def prepare_train_data(mode_id) -> dict[str, list]: def prepare_train_data(mode_id) -> dict[str, list]:
# 训练集/测试集 实体类定义, 查询条件定义
train_we, train_cd = tuple(), tuple() train_we, train_cd = tuple(), tuple()
val_we, val_cd = tuple(), tuple() val_we, val_cd = tuple(), tuple()
if mode_id == 1: if mode_id == 1:
@@ -98,8 +99,11 @@ left_col, right_col = st.columns([3, 1])
def train(): def train():
right_col.json({k: st.session_state[k] for k in ('evolve_r', 'n_trail', 'n_epoch', 'ckpt_path', 'mode')}) train_config_dict = {
right_col.json(prepare_train_data(st.session_state.mode)) k: st.session_state[k] for k in ('evolve_r', 'n_trail', 'n_epoch', 'ckpt_path', 'mode')
}
train_config_dict.update(prepare_train_data(st.session_state.mode))
right_col.json(train_config_dict)
def table_update_handler(): def table_update_handler():