完成v1, 整理好所有参数

This commit is contained in:
leo
2024-03-07 23:15:38 +08:00
parent f22d134fb4
commit 7128d489da

View File

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