morphoclass.transforms.edge_features.extract_edge_features module¶
Implementation of the base class for edge feature extractors.
-
class
morphoclass.transforms.edge_features.extract_edge_features.ExtractEdgeFeatures¶ Bases:
abc.ABCBase class for edge feature extractors.
-
abstract
extract_edge_features(data)¶ Extract some edge features from given data sample.
- Parameters
data (torch_geometric.data.Data) – A data sample.
- Returns
edge_attr – The extracted edge attributes, shape (n_edges, n_features).
- Return type
torch.tensor
-
abstract