Fix embedding
Fix various bugs in the embedding.
- Allow for missing
x
inbatch
: use other variables instead of just always relying onx
to be there - Properly get the features for the embedding
- Fix some formatting
- Fix some typos
Edited by Anthony Correia