for sample in ds:
# 还原能量矩阵
energy = np.array(sample["energy"], dtype=np.float32).reshape(
sample["n_freqs"], sample["n_dirs"]) # (47, 72)
# 锚点参数
anchors =… See the full description on the dataset page:
https://huggingface.co/datasets/wuff-mann/ATLAS-WDS.