thanks for your impl, and I found that you may not code emdedding part, I think the part is important too. I suggest that you could write a demo from input to prediction if you have free time, excespecially how to code embedding part.