Method node

graphlearn/python/graph.py:200–227

Add graph vertices that will be loaded from a given path. Args: source (string): Data source path where to load the nodes. node_type (string): Indicates the type of the added nodes. decoder (Decoder): A Decoder object to describe the data schema. mask (TRAIN | TEST | VA

(self,
           source,
           node_type,
           decoder,
           option=None,
           mask=utils.Mask.NONE)

    return data.Decoder(weighted, labeled, attr_types)

  def node(self,
           source,
           node_type,
           decoder,
           option=None,
           mask=utils.Mask.NONE):
    """ Add graph vertices that will be loaded from a given path.

    Args:
      source (string): Data source path where to load the nodes.
      node_type (string): Indicates the type of the added nodes.
      decoder (Decoder): A Decoder object to describe the data schema.
      mask (TRAIN | TEST | VAL): Mark the source as TRAIN data, TEST data
        or VAL data for the given node_type in the graph.
    """
    if not isinstance(source, str):
      raise ValueError('source for node() must be string.')
    if not isinstance(node_type, str):
      raise ValueError('node_type for node() must be string.')
    if not isinstance(decoder, data.Decoder):
      raise ValueError('decoder must be an instance of `Decoder`, got {}'
                       .format(type(decoder)))

    node_type = utils.get_mask_type(node_type, mask)
    self._node_decoders[node_type] = decoder
    node_source = self._construct_node_source(source, node_type, decoder, option)
    self._node_sources.append(node_source)
    return self

  def _copy_node_source(self, node):
    result = pywrap.NodeSource()