README.md 929 Bytes
Newer Older
chenbin zhang's avatar
chenbin zhang committed
# 代码文件说明
- analyse.ipynb
    - Case study + visualization
    - 生成一些结果表格
- get_dockQ.py
    - 根据模型预测结果计算dockQ,将具体的评测结果存储在`scores/*.csv`中,将总结的评测结果存储在`scores/*.json`
- draw.py
    - 画图
    - 使用方式`python draw.py json_path_1 ... json_path_n jpg_path
- batch_draw.sh
    - 批量画图
- ensemble_dockQ.py
    - 输入多个个csv文件,根据csv文件中的pred_path找到conf_path,然后从多个模型结果中选择confidence最高的
    - 使用方式`python ensemble_dockQ.py csv_path_1, ..., csv_path_n save_prefix ranking_metric
        - save_prefix表示存储文件的前缀,该代码最后会存储成f'{save_prefix}.csv'和f'{save_prefix}.json'
        - ranking_metric取iptm或ranking_score
- sota_draw.py
    - 选出最好的两个结果画图
- utils.py
    - 字符串匹配的动态规划算法