contextual features set, working training properly ( negative likelihood and all)