Skip to content
Snippets Groups Projects
Commit 42468af2 authored by aBotel's avatar aBotel
Browse files

Update BERT xform to return a tensor (#cond+#proc+#drug, 768) i.e. seq

of codes. Need seq for RNN instead of summing.

Add 2 collate functions 1 for CodeEmb, 1 for DescEmb.

Will need each of them to pad/reverse seq to max seq length.
parent 2a43602f
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment