Views
No views yet
1from nexamat import GNNPredictor, VAEMaterialGenerator
2
3# Initialize models
4predictor = GNNPredictor.load("Allanatrix/predictor.pt")
5vae = VAEMaterialGenerator.load("Allanatrix/vae.pt")
6
7# Predict properties for a material
8material_graph = load_material("LiFePO4.json")
9prediction = predictor(material_graph)
10
11# Generate novel material candidates
12latent_sample = vae.sample_latent()
13generated_material = vae.decode(latent_sample)