The NodePieceMLPTransform converts a batch of data from the NodePieceLoader into a format that can be used in a MLP implemented in PyTorch.
init(label: str, features: list, target_type: str = None)
Instantiate a NodePieceMLPTransform.
The attribute name that corresponds with the label being predicted. Supports numerical datatypes (INT, FLOAT, DOUBLE).
features (list of str, optional): List of attributes to use as features into the model. Supports numerical datatypes (INT, FLOAT, DOUBLE).
target_type (str, optional): The type of vertex to perform predictions on.