how to visualize the generated results?

[yuuhaixia]

Thank you very much for your work, after training the generation task, may I ask how to visualize the generated results?

[JianZhou711]

You have to take the Pytorch file and convert the data using the Crystal class in the code which you can then feed into packages like ASE, nglview, or some other software.

[yuuhaixia]

class Crystal(object):

def __init__(self, crys_array_dict):
    self.frac_coords = crys_array_dict['frac_coords']
    self.atom_types = crys_array_dict['atom_types']
    self.lengths = crys_array_dict['lengths']
    self.angles = crys_array_dict['angles']
    self.dict = crys_array_dict

...........

Hello, are you referring to this Crystal class below in the compute_metrics.py file? Do I need to run through this evaluation file to visualize my generated results?

If I just run python cdvae/run.py data=perov expname=perov
and
python scripts/evaluate.py --model_path MODEL_PATH --tasks recon gen

Is it possible to get the results of the generated crystals, as well as visualize them?

[yuuhaixia]

@JianZhou711

[JianZhou711]

Our workflow included importing the crystal class and using it to convert the PT file then converting the info from that into a poscar before visualizing it with Vesta (example in the attached image). I have experimented with running it on ASE and NGLview without converting poscar and have worked, but I do not have those files atm. There might be a way to do it without the class, but we just determined that was the easiest method.

[yuuhaixia]

Thank you very much for your reply, can you please share the code to convert PT to POSCAR, I keep having problems! @JianZhou711

[yuuhaixia]

The output of the PT in the generated material does not tell the type of atom, such as whether it is Bi or O, etc., so how should one determine what component the generated material is?

[JianZhou711]

https://www.dropbox.com/scl/fo/2jrwuphiulv5rkgji8der/AAM_BCKXbj_nbGSmz7owoPc?rlkey=w6hg37silgt4eboooqy4im4jm&st=ydbhn7f5&dl=0

This should answer both questions. I'll leave it up for a few days before deleting it.

[yuuhaixia]

Thank you really, but I don't understand why I'm getting this error AttributeError: 'Crystal' object has no attribute 'structure'

[yuuhaixia]

@JianZhou711

[JianZhou711]

That's weird, we never ran into that error when doing test runs with mp-20. Try doing a try-except run instead of a list comprehension and skip all of the ones that ran into an error.

[yuuhaixia]

Is it something to do with the number of times it was trained, because I only trained 500 epochs, and the output 'atom_types' are all negative. Below is the result of my PT printout, is there any anomaly?
Thanks again！

@JianZhou711

[JianZhou711]

Potentially? I didn't see those issues when I ran, but I did mainly use the mp_20 dataset for trains. I could ask others, but I'm on break for a few weeks. I would recommend just running try excepts and ignoring them before doing another model run to test.

[yuuhaixia]

If I skip the exception then I will get null value. Can you provide me with a PT file that you have trained and generated, I want to make sure that there is something wrong with my generated data that is causing the visualization not to work. Really thank you very much for replying me multiple times~

@JianZhou711

[JianZhou711]

same dropbox

[yuuhaixia]

Many thanks to the authors for their contribution and sharing. However, when I am using the PT file you shared for visualization, (I am using VESTA), one of the crystal diagrams I get is as follows: there is no chemical bond link between atoms. May I ask why this is so? Or is there something wrong with @JianZhou711