Training config for the DOCUMENT_SUMMARIZATION problem type
KEY | TYPE | Description |
---|---|---|
DROPOUT_RATE | float | Dropout rate for neural network. |
BATCH_SIZE | BatchSize | Batch size for neural network. |
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. |