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

test_params

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)