Encoder forward pass :param inputs: shape (seq_len, batch_size, num_sensor * input_dim) :return: encoder_hidden_state: (num_layers, batch_size, self.hidden_state_size)
(self, inputs, adj)
| 166 | self.cl_decay_steps + np.exp(batches_seen / self.cl_decay_steps)) |
| 167 | |
| 168 | def encoder(self, inputs, adj): |
| 169 | """ |
| 170 | Encoder forward pass |
| 171 | :param inputs: shape (seq_len, batch_size, num_sensor * input_dim) |
| 172 | :return: encoder_hidden_state: (num_layers, batch_size, self.hidden_state_size) |
| 173 | """ |
| 174 | encoder_hidden_state = None |
| 175 | for t in range(self.encoder_model.seq_len): |
| 176 | _, encoder_hidden_state = self.encoder_model(inputs[t], adj, encoder_hidden_state) |
| 177 | |
| 178 | return encoder_hidden_state |
| 179 | |
| 180 | def decoder(self, encoder_hidden_state, adj, labels=None, batches_seen=None): |
| 181 | """ |