Skip to content

xuqch/TabPFN_code

Repository files navigation

主要文件

  • clf.sh:运行程序的控制文件,通过参数设定运行不同的程序
  • main_tabpfn_clf_reg.py:主控制文件,通过参数读取运行不同的代码块
  • config.py:参数设定文件,所有的默认参数均保存于此,包括数据默认路径、空间范围设置等
  • init.py:Python包初始化文件

处理流程

Step1:模型训练与站点预测
  • 模型训练:训练代码位于 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.py
  • Step1_Fig_result.py
Step2:格点预测
  • 通过 clf.sh 文件中的参数设定进入该程序
  • 预测代码位于 Step2_main_tabpfn_grid.py
  • 前置数据处理代码通过 Step2_Process_grid_input.py 保存在npy文件中,并在预测程序中重新加载

相关文件:

  • Step2_main_tabpfn_grid.py
  • Step2_Process_grid_input.py
Step3:后处理步骤
  • Step3_Process.py:后处理程序(开发中)

基本使用

  1. 配置参数: 首先在 config.py 中设置相关参数,包括数据路径、空间范围等。

  2. 运行站点训练与预测

    # 通过 clf.sh 运行

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published