graphnet.py
:神经网络的构造、训练、测试代码graphdata.py
:数据预处理的相关代码torchamg.py
:基于pytorch
实现的迭代算法main.py
:程序入口TestPyAMG/
:用于测试pyamg
的目录TestSparseTensor/
:用于测试sparse tensor
的目录
以上代码是其它项目的代码,还需要针对本项目进行修改。运行以上代码需要安装pytorch geometric
在verify_by_pyamg.py
中使用pyamg
中的两网格算法测试矩阵p
的求解效率。为了提高直接法的运行效率,需要安装umfpack
求解器,通过以下命令进行安装
conda install -c conda-forge suitesparse
conda install scikit-umfpack
可以修改verify_by_pyamg.py
中磨光算法和粗网格上的求解算法,部分可选参数在注释行中。详细设置可参考 pyamg.multilevel,磨光算法和求解算法的选择没有完善的文档,只能在程序源码中寻找。