Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Graph data structure update and inframe batchlization #44

Closed
wants to merge 92 commits into from

Conversation

floatingCatty
Copy link
Member

@floatingCatty floatingCatty commented Nov 3, 2023

  1. Add graph data structure from NequIP, adapt to generalize edge feature/ eigenvalues/ kpoints data
  2. Add graph network class from NequIP
  3. Implementing e3 hamiltonian constructor module and its interfaces with SKTB rotation.
  4. Implementing indexmap for e3 module, which is for later usage.

@QG-phy QG-phy added the Feature Request New feature or request label Nov 9, 2023
dptb/nn/nnsk.py Outdated Show resolved Hide resolved
dptb/nn/nnsk.py Outdated Show resolved Hide resolved
floatingCatty and others added 2 commits January 18, 2024 14:14
* new version of `E3statistics` function added in DefaultDataset.

* fix bug in dealing with scalars in `E3statistics`

* add "decay" option in E3statistics to return edge length dependence

* fix bug in getting rmes when doing stat & update argcheck
dptb/nn/deeptb.py Outdated Show resolved Hide resolved
dptb/data/transforms.py Outdated Show resolved Hide resolved
dptb/nn/nnsk.py Outdated Show resolved Hide resolved
@QG-phy QG-phy closed this Feb 2, 2024
@floatingCatty floatingCatty deleted the data_nequip branch March 28, 2024 10:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Feature Request New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants