adjusted session size prediction, fixed embeddings

adjusted session size prediction, fixed embeddings #206