Better parameterize PartOfSpeechTagLayer.