class_docstr: Automagic DGL models from Graphistry Instances.
class_docstr_tok: ["Automagic","DGL","models","from","Graphistry","Instances","."]
code: def _mask_nodes(self):
      if config.FEATURE in self._dgl_graph.ndata:
          n = self._dgl_graph.ndata[config.FEATURE].shape[0]
          (
              self._dgl_graph.ndata[config.TRAIN_MASK],
              self._dgl_graph.ndata[config.TEST_MASK],
          ) = get_torch_train_test_mask(n, self.train_split)
  
imports: from collections import Counter
  from typing import Dict, Optional, TYPE_CHECKING, Tuple
  import numpy
  import pandas
  from .None import constants
  from .feature_utils import FeatureEngine, FeatureMixin, resolve_feature_engine, XSymbolic, YSymbolic, resolve_X, resolve_y
  from .util import setup_logger
label_desc: Use image node_id 11 for calling the DGLGraphMixin obj's underlying member method code with example usage: obj._mask_nodes() without return types