Fix embedding
Fix various bugs in the embedding.
- Allow for missing
xinbatch: use other variables instead of just always relying onxto be there - Properly get the features for the embedding
- Fix some formatting
- Fix some typos
Fix various bugs in the embedding.
x in batch: use other variables instead of just always relying on x to be there