Indexing an array 'A' with shape (n, self.n_nodes, out-dim) leads to A[:, get_level_nodes_mask(), :] have shape (n, len(self.LEVEL_NO

Plot MAE and MBE as a function of the height/pressure :param Ytrue: :param preds: :param height_ticks: must have same shape as Ytrue.

For Graph networks :param adj: A N x N adjacency matrix :return: In-degree normalized matrix (Row-normalized matrix)

Either name_to_array dict OR ALL OF global_node, layers mustnt be None

This method controls which parameters from Hydra config are saved by Lightning loggers. Credits go to: https://github.com/ashleve/lightning-hydra-

Returns: The number of feature dimensions that the transformed data will have. If the transform returns an array, out

Download data if needed. This method is called only from a single GPU. Do not use it to assign state (self.x = y).

Prints content of DictConfig using Rich library and its tree structure. Credits go to: https://github.com/ashleve/lightning-hydra-template A

Load data. Set internal variables: self._data_train, self._data_val, self._data_test.