search

woshiyyya / ERPP-RMTPP

A pytorch implementation of ERPP and RMTPP on ATM maintenance dataset.
55 stars 21 forks source link

Forward pass of ERPP #12

Closed HaochenWang1243 closed 6 months ago

HaochenWang1243 commented 6 months ago

Hi. I think in the ERPP paper, the author fused the output of the 2 LSTMs into a single embedding vector. However when I'm looking at model.py: image This seems to only embed the output of the classification LSTM? Thanks.

HaochenWang1243 commented 6 months ago

NVM! Realized this implementation is correct.