gobbli.test.classification.test_embeddings module

gobbli.test.classification.test_embeddings.check_embed_output(embed_input, embed_output, expected_dimensionality=None, max_seq_length=None)[source]

Verify some information regarding the embedding output. User can optionally pass expected values of the max sequence length and dimensionality, if they’re known. If not, we’ll just verify the dimensionality is larger than 0.

gobbli.test.classification.test_embeddings.test_embeddings(model_cls, dataset_cls, model_kwargs, train_kwargs, embed_kwargs, model_gpu_config, pooling, gobbli_dir, request)[source]

Ensure embedding models train and generate embeddings appropriately across a few example datasets.