-
Notifications
You must be signed in to change notification settings - Fork 9
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
[WIP] Implementation of 2D crystal former #7
base: main
Are you sure you want to change the base?
Conversation
Layer group functions and corresponding tests added. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Tests are well written 👍🏾 . Please clean up the repo.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please remove.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please remove.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
If this dataset is for testing, maybe move it to the test folder?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
What is the difference between these two files? Why not convert the items to list in a programmatical way?
from crystalformer.src.lattice import symmetrize_lattice | ||
from crystalformer.src.wyckoff import mult_table, symops | ||
from crystalformer.src.sym_group import * | ||
# from crystalformer.src.lattice import symmetrize_lattice |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove.
|
||
import numpy as np | ||
np.set_printoptions(threshold=np.inf) | ||
# import numpy as np |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove this file.
x = jnp.array([0.123, 0.123, 0.25]) | ||
y = project_x(g, w, x, idx) | ||
assert jnp.allclose(y, jnp.array([0.877, 0.877, 0.75])) | ||
# x = jnp.array([0.123, 0.123, 0.25]) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove.
|
||
# def test_symmetrize_atoms(): | ||
# from crystalformer.src.wyckoff import symmetrize_atoms | ||
# from pymatgen.symmetry.groups import SpaceGroup |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove.
|
||
|
||
# ============= params to control the generation ============= | ||
spacegroup = 2 # 设置生成的晶体的空间群,范围为1-230 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
No chinese in comments.
This PR is not ready yet.