Family and Polarity Tagger question

[jeffmax]

Should https://github.com/som-shahlab/ehr-rwe/blob/master/rwe/labelers/taggers/family.py#L128 be set to

self.class_map[y]

instead of the index number y?

Similarly, for the polarity tagger here:

https://github.com/som-shahlab/ehr-rwe/blob/master/rwe/labelers/taggers/polarity.py#L532

Thanks for the great library.