That’s good to hear! But I am not sure I understand it. If I do not feed partial outputs at prediction time how do I control which piece of information I want to extract from the document? Naively I would think in this case the output will always be the most frequent type of information in the dataset (age, name or whatever). So for example it will always be “name: George” because some documents in the training set only contained name but no age thus name is the most likely field to generate.