class_docstr: GATs Model
  
  Parameters
  ----------
  lr : float
      learning rate
  d_feat : int
      input dimensions for each time step
  metric : str
      the evaluation metric used in early stop
  optimizer : str
      optimizer name
  GPU : int
      the GPU ID used for training
code: def __init__(
      self,
      d_feat=20,
      hidden_size=64,
      num_layers=2,
      dropout=0.0,
      n_epochs=200,
      lr=0.001,
      metric="",
      early_stop=20,
      loss="mse",
      base_model="GRU",
      model_path=None,
      optimizer="adam",
      GPU=0,
      n_jobs=10,
      seed=None,
      **kwargs,
  ):
      # Set logger.
      self.logger = get_module_logger("GATs")
      self.logger.info("GATs pytorch version...")
  
      # set hyper-parameters.
      self.d_feat = d_feat
      self.hidden_size = hidden_size
      self.num_layers = num_layers
      self.dropout = dropout
      self.n_epochs = n_epochs
      self.lr = lr
      self.metric = metric
      self.early_stop = early_stop
      self.optimizer = optimizer.lower()
      self.loss = loss
      self.base_model = base_model
      self.model_path = model_path
      self.device = torch.device(
          "cuda:%d" % (GPU)
          if torch.cuda.is_available() and GPU >= 0
          else "cpu"
      )
      self.n_jobs = n_jobs
      self.seed = seed
  
      self.logger.info(
          "GATs parameters setting:"
          "\nd_feat : {}"
          "\nhidden_size : {}"
          "\nnum_layers : {}"
          "\ndropout : {}"
          "\nn_epochs : {}"
          "\nlr : {}"
          "\nmetric : {}"
          "\nearly_stop : {}"
          "\noptimizer : {}"
          "\nloss_type : {}"
          "\nbase_model : {}"
          "\nmodel_path : {}"
          "\nvisible_GPU : {}"
          "\nuse_GPU : {}"
          "\nseed : {}".format(
              d_feat,
              hidden_size,
              num_layers,
              dropout,
              n_epochs,
              lr,
              metric,
              early_stop,
              optimizer.lower(),
              loss,
              base_model,
              model_path,
              GPU,
              self.use_gpu,
              seed,
          )
      )
  
      if self.seed is not None:
          np.random.seed(self.seed)
          torch.manual_seed(self.seed)
  
      self.GAT_model = GATModel(
          d_feat=self.d_feat,
          hidden_size=self.hidden_size,
          num_layers=self.num_layers,
          dropout=self.dropout,
          base_model=self.base_model,
      )
      self.logger.info("model:\n{:}".format(self.GAT_model))
      self.logger.info(
          "model size: {:.4f} MB".format(
              count_parameters(self.GAT_model)
          )
      )
  
      if optimizer.lower() == "adam":
          self.train_optimizer = optim.Adam(
              self.GAT_model.parameters(), lr=self.lr
          )
      elif optimizer.lower() == "gd":
          self.train_optimizer = optim.SGD(
              self.GAT_model.parameters(), lr=self.lr
          )
      else:
          raise NotImplementedError(
              "optimizer {} is not supported!".format(optimizer)
          )
  
      self.fitted = False
      self.GAT_model.to(self.device)
  
imports: from __future__ import division
  from __future__ import print_function
  import numpy
  import pandas
  import copy
  from ...utils import get_or_create_path
  from ...log import get_module_logger
  import torch
  import torch.nn
  import torch.optim
  from torch.utils.data import DataLoader
  from torch.utils.data import Sampler
  from .pytorch_utils import count_parameters
  from ...model.base import Model
  from ...data.dataset.handler import DataHandlerLP
  from ...contrib.model.pytorch_lstm import LSTMModel
  from ...contrib.model.pytorch_gru import GRUModel
label_desc: Use image node_id 1 to create a new GATs object from inherited base classes: Model with example: obj = GATs(d_feat, hidden_size, num_layers, dropout, n_epochs, lr, metric, early_stop, loss, base_model, model_path, optimizer, GPU, n_jobs, seed)