Optum/retain-keras

Question about 'to_event' formatting

jroberayalas opened this issue · 2 comments

jroberayalas commented

Hello! Quick question: when you say:

Column 'to_event' is a list of values indicating when the respective visit happened. Values have to be ordered from oldest to newest.

Are these values a pd.Timestamp() containing the date of the visit?

tRosenflanz commented

No, they should be some numeric value. I usually use days to index event or days since last time step - either will work. You can check appendix A of original RETAIN paper for some more details

  • 👍1
jroberayalas commented

Great! Thanks for the clarification.