- clf.sh:运行程序的控制文件,通过参数设定运行不同的程序
- main_tabpfn_clf_reg.py:主控制文件,通过参数读取运行不同的代码块
- config.py:参数设定文件,所有的默认参数均保存于此,包括数据默认路径、空间范围设置等
- init.py:Python包初始化文件
- 模型训练:训练代码位于
main_tabpfn_clf_reg.py中的main_RF_process函数,完成训练后会调用可视化代码 - 站点预测:代码位于
main_tabpfn_clf_reg.py中的site_predict函数,可视化代码在Step1_Fig_result.py - 数据处理:训练和站点预测的输入数据处理位于
Step1_Process_dataframe.py文件中,通过参数生成对应时间范围的npy文件,并依据KGE_threshold或used_index挑选站点进行训练与预测
相关文件:
Step1_Process_dataframe.pyStep1_Fig_result.py
- 通过
clf.sh文件中的参数设定进入该程序 - 预测代码位于
Step2_main_tabpfn_grid.py - 前置数据处理代码通过
Step2_Process_grid_input.py保存在npy文件中,并在预测程序中重新加载
相关文件:
Step2_main_tabpfn_grid.pyStep2_Process_grid_input.py
- Step3_Process.py:后处理程序(开发中)
-
配置参数: 首先在
config.py中设置相关参数,包括数据路径、空间范围等。 -
运行站点训练与预测:
# 通过 clf.sh 运行