DocumentClassificationTrainingConfig

Training config for the DOCUMENT_CLASSIFICATION problem type

KEY TYPE Description
ZERO_SHOT_HYPOTHESES List[str] Zero shot hypotheses. Example text: 'This text is about pricing'.
TEST_SPLIT int Percent of dataset to use for test data. We support using a range between 5 ( i.e. 5% ) to 20 ( i.e. 20% ) of your dataset.