gigl.src.common.utils.data.training#
Functions#
|
Returns tensor with features from x based on feature_names |
Module Contents#
- gigl.src.common.utils.data.training.filter_features(feature_schema, feature_names, x)[source]#
Returns tensor with features from x based on feature_names
- Parameters:
feature_schema (gigl.src.data_preprocessor.lib.types.FeatureSchema)
feature_names (list[str])
x (torch.Tensor)
- Return type:
torch.Tensor