DocumentVisualizationTrainingConfig

Training config for the DOCUMENT_VISUALIZATION 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.