Simplify subsampling logic

Store the probability for each vocab word, regardless of the value of `self.sample`.

This makes the subsequent access to `vocab.sample_probability` more uniform and cleaner.