fix lemma, option for class weight