code: def __init__(self):
      super().__init__()
      self.max_size = (MemoryGranularity.Transitions, 1000000)
      self.allow_duplicates_in_batch_sampling = False
      self.num_classes = 0
      self.state_key_with_the_class_index = "class"
  
imports: import operator
  import random
  from enum import Enum
  from typing import List, Tuple, Any, Union
  import numpy
  from rl_coach.core_types import Transition
  from rl_coach.memories.memory import MemoryGranularity
  from rl_coach.memories.non_episodic.experience_replay import ExperienceReplayParameters, ExperienceReplay
  from rl_coach.schedules import Schedule, ConstantSchedule
label_desc: Use image node_id 1 to create a new BalancedExperienceReplayParameters object from inherited base classes: ExperienceReplayParameters with example: obj = BalancedExperienceReplayParameters()