Skip to content

Commit

Permalink
reandme and tutorial
Browse files Browse the repository at this point in the history
  • Loading branch information
killiansheriff committed Mar 11, 2024
1 parent 31afea3 commit 129610a
Show file tree
Hide file tree
Showing 2 changed files with 388 additions and 40 deletions.
24 changes: 0 additions & 24 deletions examples/_frameworks.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,27 +111,3 @@ def import_model_config(self):
"mul": 50, # 50
}
self.model_config = model_config



def __init__(self, **kwargs):
super().__init__(**kwargs)

def import_model_config(self):
lmax, layers, outlength, number = 2, 2, 4, 0 # 2,2,4,0

model_config = {
"out_feature_length": outlength,
"max_radius": 3, # 2.5
"min_radius": 0,
"number_of_basis": 10,
"num_nodes": 12, # only used for feqtur enormalization, we don't really care set it to a constant so that we can have only 1 network for all the crystal structure
"num_neighbors": 5, # only used for feqtur enormalization, we don't really care set it to a constant so that we can have only 1 network for all the crystal structure
"layers": layers,
"lmax": lmax,
"net_number": number,
"irreps_node_attr": "5x0e",
"model_load": f"/home/ksheriff/PAPERS/second_paper/02_1nn_synthetic/data/nets/net_{lmax}-{layers}-{outlength}_{number}.pt",
"mul": 3, # 50
}
self.model_config = model_config
404 changes: 388 additions & 16 deletions examples/fcc_ternary_system.ipynb

Large diffs are not rendered by default.

0 comments on commit 129610a

Please sign in to comment.