class_docstr: RotatE model.
class_docstr_tok: ["RotatE","model","."]
code: def __init__(
      self,
      data_reader,
      hidden_size,
      margin,
      learning_rate,
      args,
      optimizer="adam",
  ):
      super(RotatE, self).__init__(
          model_name="RotatE",
          data_reader=data_reader,
          hidden_size=hidden_size,
          margin=margin,
          learning_rate=learning_rate,
          args=args,
          optimizer=optimizer,
      )
  
      self._neg_times = self.args.neg_times
      self._adv_temp_value = self.args.adv_temp_value
  
      self._relation_hidden_size = self._hidden_size
      self._entity_hidden_size = self._hidden_size * 2
      self._entity_embedding_margin = (
          self._margin + 2
      ) / self._entity_hidden_size
      self._relation_embedding_margin = (
          self._margin + 2
      ) / self._relation_hidden_size
      self._rel_shape = [
          self._relation_total,
          self._relation_hidden_size,
      ]
      self._ent_shape = [self._entity_total, self._entity_hidden_size]
      self._pi = 3.141592654
  
      self.construct_program()
  
file_path: PGL/legacy/examples/pgl-ke/model/RotatE.py
imports: import paddle.fluid
  from .Model import Model
  from .utils import lookup_table
label_desc: Use image node_id 1 to create a new RotatE object from inherited base classes: Model with example: obj = RotatE(data_reader, hidden_size, margin, learning_rate, args, optimizer)