I want to create a model where a user can define a set of labels for the purpose of classification of documents. Each label for that document should also have a probability