core.datasets._utils#
Copyright (c) Meta, Inc. and its affiliates.
This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.
Functions#
|
Rename data object keys |
Module Contents#
- core.datasets._utils.rename_data_object_keys(data_object: torch_geometric.data.Data, key_mapping: dict[str, str | list[str]]) torch_geometric.data.Data #
Rename data object keys
- Parameters:
data_object – data object
key_mapping – dictionary specifying keys to rename and new names {prev_key: new_key}
keys (new_key can be a list of new)
example (for)
:param : :param prev_key: energy :param new_key: [common_energy, oc20_energy] :param This is currently required when we use a single target/label for multiple tasks: