biotransformers.tests.test_embeddings

Contents

`biotransformers.tests.test_embeddings`¶

Test module for testing embeddings function

Module Contents¶

Functions¶

`test_embeddings_type_and_shape`(init_model, sequences, batch_size, pool_mode)
`test_embeddings_type_and_shape_fasta`(init_model, fasta_path, lengths_sequence_fasta, batch_size, pool_mode)

Attributes¶

biotransformers.tests.test_embeddings.test_params = [[1, ['cls', 'mean']], [2, ['full', 'mean', 'cls']], [10, ['cls', 'full']]]¶

biotransformers.tests.test_embeddings.test_embeddings_type_and_shape(init_model, sequences, batch_size, pool_mode)¶

biotransformers.tests.test_embeddings.test_embeddings_type_and_shape_fasta(init_model, fasta_path, lengths_sequence_fasta, batch_size, pool_mode)¶