learn.property_prediction.build_dataset#
- build_dataset(targets, graph_dataset)[source]#
Builds dataset in format that is used for data loading and throughout property predictions.
- Parameters:
targets – Dict containing all targets to be predicted. Can be retrieved in error_fn under the respective key.
graph_dataset – Dataset of graphs, e.g. as obtained from
jax_md_mod.model.sparse_graph.convert_dataset_to_graphs().
- Returns:
A dictionary containing the combined dataset and a list of target keys