class_docstr: Manages the partitioning related controls.
  
  Handles data generation and informs the model about changes in partition.
class_docstr_tok: ["Manages","the","partitioning","related","controls",".","Handles","data","generation","and","informs","the","model","about","changes","in","partition","."]
docstr: Initialize the Partitioning Data Manager.
  
  Uses/Creates partitioner and generates partition related parameters.
  
  Parameters
  ----------
  dataset_loader : AbstractGraphPartitioner or GraphDataLoader
      Either an instance of AbstractGraphPartitioner or GraphDataLoader.
  model: tf.keras.Model
      The model that is being trained.
  strategy: str
      Type of partitioning strategy to use.
  root_directory: str
      Directory where the partition manager files will be stored.
code: def __init__(
      self,
      dataset_loader,
      model,
      strategy="RandomEdges",
      partitioner_k=3,
      root_directory=None,
  ):
      """Initialize the Partitioning Data Manager.
  
      Uses/Creates partitioner and generates partition related parameters.
  
      Parameters
      ----------
      dataset_loader : AbstractGraphPartitioner or GraphDataLoader
          Either an instance of AbstractGraphPartitioner or GraphDataLoader.
      model: tf.keras.Model
          The model that is being trained.
      strategy: str
          Type of partitioning strategy to use.
      root_directory: str
          Directory where the partition manager files will be stored.
      """
      super(GeneralPartitionDataManager, self).__init__(
          dataset_loader, model, strategy, partitioner_k, root_directory
      )
  
imports: import abc
  import glob
  import logging
  import os
  import shelve
  import shutil
  import tempfile
  from datetime import datetime
  import numpy
  import tensorflow
  from .graph_partitioner import PARTITION_ALGO_REGISTRY, AbstractGraphPartitioner
label_desc: Use image node_id 1 to create a new GeneralPartitionDataManager object from inherited base classes: PartitionDataManager with example: obj = GeneralPartitionDataManager(dataset_loader, model, strategy, partitioner_k, root_directory)