'Returns the type with the highest probability for the given document.
Returns a dict of (class, probability)-items if discrete=False.'

def classify(self, document, discrete=True):

    if (not discrete):
        return Probabilities(self, {})
    return self.baseline
