nsl.lib.strip_neighbor_features

View source on GitHub

Strips graph neighbor features from a feature dictionary.

features Dictionary of tensors mapping feature names to tensors. This dictionary includes sample features but may or may not include corresponding neighbor features for each sample feature.
neighbor_config An instance of nsl.configs.GraphNeighborConfig.

A dictionary mapping only sample feature names to tensors. Neighbor features in the input are not included.