moved to a custom label encoder, which also seems to have drastically improved scores