Skip to main content

map_to_graph_keys

Function map_to_graph_keys 

Source
pub fn map_to_graph_keys(table: &WeightTable) -> BTreeMap<String, String>
Expand description

Translation table — rewrite DeePMD’s serialized param paths into the crate::descriptor::* / crate::fitting::* param-name conventions. Both se_e2_a (type_one_side = true) and the shared-net energy fitting are covered; extending this map is the usual integration step when a new descriptor / fitting variant becomes available.

Returned map: graph-param-name → deepmd-key (so callers can .remove the deepmd key from WeightTable and bind by graph name).