Json数据

json数据的键必须是元素名,例如 {“H”: … ,”He”: … }, 这里的结构是 pymatgen Structure 列表

  • 根据结构索引。

>>> from featurebox.featurizers.atom.mapper import AtomJsonMap
>>> tmps = AtomJsonMap(search_tp="number", embedding_dict="ele_megnet.json")
>>> a = tmps.convert(structurei)

返回的数据是1,76号元素的属性。

  • 用自己的json数据,按数字索引。

>>> from featurebox.featurizers.atom.mapper import AtomJsonMap
>>> tmps = AtomJsonMap(search_tp="number",embedding_dict="ele_megnet.json")
>>> s = [1,76]
>>> a = tmps.convert(s)

返回的数据是1,76号元素的属性。

../_images/1_1.png
  • 根据字典索引。

>>> from featurebox.featurizers.atom.mapper import AtomJsonMap
>>> tmps = AtomJsonMap(search_tp="name")
>>> s = [{"H": 2, }, {"Al": 1}] # or [{i.element.symbol:1} for i in structure.species]
>>> a = tmps.convert(s)
../_images/1_3.png
  • 批量数据。

>>> from featurebox.featurizers.atom.mapper import AtomJsonMap
>>> tmps = AtomJsonMap(search_tp="name")
>>> s = [[{"H": 2, }, {"Ce": 1}],[{"H": 2, }, {"Al": 1}]]
>>> a = tmps.transform(s)

默认返回数据为np.ndarray类型。

../_images/1_2.png ../_images/1_3.png