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

完整数据展示

Browse Source
This commit is contained in:
leo
2024-03-07 20:26:09 +08:00
parent 8c78592f0a
commit 8d0e6ef4dd
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@@ -22,6 +22,7 @@ st.set_page_config(
initial_sidebar_state="expanded", initial_sidebar_state="expanded",
) )
@st.cache_data @st.cache_data
def list_ckpt_paths(dir_path): def list_ckpt_paths(dir_path):
return os.listdir(dir_path) return os.listdir(dir_path)
@@ -64,6 +65,7 @@ def update_handler():
row_id = int(edited_df.loc[id_].id) row_id = int(edited_df.loc[id_].id)
row_db = session.query(BatchData).where(BatchData.id == row_id).first() row_db = session.query(BatchData).where(BatchData.id == row_id).first()
logger.info(f"{row_id=}, {update_data=}") logger.info(f"{row_id=}, {update_data=}")
logger.debug(BatchDataRead.from_orm(row_db))
for field in update_data: for field in update_data:
setattr(row_db, field, update_data[field]) setattr(row_db, field, update_data[field])
session.commit() session.commit()
@@ -82,7 +84,8 @@ with left_col:
hide_index=True, hide_index=True,
use_container_width=True, use_container_width=True,
on_change=update_handler, on_change=update_handler,
column_order=('id', 'year', 'census_batch', 'id_code', 'precision', 'is_train', 'is_validation', 'ann_file', 'img_prefix'), column_order=(
'id', 'year', 'census_batch', 'id_code', 'precision', 'is_train', 'is_validation', 'ann_file', 'ann_file_lbs', 'img_prefix',),
column_config={ column_config={
"year": st.column_config.NumberColumn("年份", format="%d 年", ), "year": st.column_config.NumberColumn("年份", format="%d 年", ),
'census_batch': "普查批次", 'census_batch': "普查批次",
@@ -90,8 +93,9 @@ with left_col:
'precision': "精度", 'precision': "精度",
'is_train': "是否是训练集", 'is_train': "是否是训练集",
'is_validation': "是否是验证集", 'is_validation': "是否是验证集",
'ann_file': "path", 'ann_file': st.column_config.Column("path", width='medium'),
'ann_file_lbs': "lbs_path", 'ann_file_lbs': st.column_config.Column("lbs_path", width='medium'),
'img_prefix': st.column_config.Column("img_prefix", width='medium'),
}) })
with right_col: with right_col: