Arraymancer
Arraymancer copied to clipboard
CRF head [WIP]
I'm going to need some help refining this (especially not sure yet how the backward pass will work), but I think that I can add a CRF head for sequence prediction that should work with (for example) the GRU layer.
I've been mostly following your guide https://github.com/mratsim/Arraymancer/issues/331
TODO:
- Finish forward implementation
- Backward pass implementation
- Lots more testing / move tests to test directory
- Example?
Don't hesitate to ask if you are stuck on a specific thing. I'm available here or all day through the Gitter channel or Nim IRC.
Thanks, appreciate it! I'm going to try and pick this up again tonight/tomorrow and will probably have some better questions once I'm done with the forward pass.
Haven't forgotten about this, just haven't had as much time to work on this as I thought I would.
No problem, I don't have much time myself