CustomTrainedModelTrainingConfig

Training config for the CUSTOM_TRAINED_MODEL problem type

KEY TYPE Description
MAX_DIMENSION int Maximum expected dimension of the catalog.
INDEX_OUTPUT_PATH str Fully qualified cloud location (GCS, S3, etc) to export snapshots of the embedding to.
DOCKER_IMAGE_URI str Docker image URI.
SERVICE_PORT int Service port.
MAX_CATALOG_SIZE int Maximum expected catalog size.
STREAMING_EMBEDDINGS bool Flag to enable streaming embeddings.