def METHOD_NAME(self):
    """ Check probability calculation where `use_unknown=False` """ # noqa
    data = ['category 1', 'category 3', 'category 4', None, 'category 3']
    enc = OneHotEncoder(use_unknown=False)
    enc.prepare(data)
    # Make data to represent random weights that do not sum to 1
    torch.manual_seed(1)
    wt_vec = torch.rand(size=(len(data), len(enc.map)))
    _, probs, _ = enc.decode_probabilities(wt_vec)
    self.assertTrue(np.all([np.isclose(sum(i), 1) for i in probs]))