Infinite Potential Edges #68
Open
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Infinite Potential Edges
This is a PR for adding support for infinite potential edges from https://arxiv.org/pdf/2306.10045.pdf
Beware that this does involve cython code, which means in order to use this added support, the cython code has to be compiled or there will be errors.
General
helpers.py
many changes herecalculate_inf_potentials
method which calculates the pauli, LD, and Coulomb potentials.calculate_edges_master
method has been updated to support infinite edge calculation, and create inf_edge_attr (weighted sum of potentials mentioned above) and inf_edge_indexgenerate_edge_features_inf
method concatenates the gaussian expansion of inf_edge_attr and gaussian expansion of edge weightsprocessor.py
added support for above methods in helpersconfigs/example/config_potnet.yml
example config for a cgcnn model using infinite edgesmatdeeplearn/preprocessor/inf_functions
here there is the cython helper code forcalculate_inf_potentials
module load gsl
, on your local device you need to set it up with thegsl-latest.tar.gz
file.python3 setup.py build_ext --inplace