biotransformers.tests.test_logits

Test module for testing logits function

Module Contents

Functions

test_logits_type(init_model, batch_size, sequences, pass_mode)
test_logits_type_fasta(init_model, batch_size, fasta_path, lengths_sequence_fasta, pass_mode)

Attributes

test_params
test_params_fasta

biotransformers.tests.test_logits.test_params = [[1, 'forward'], [2, 'masked'], [10, 'forward']]

biotransformers.tests.test_logits.test_params_fasta = [[2, 'forward']]

biotransformers.tests.test_logits.test_logits_type(init_model, batch_size, sequences, pass_mode)

biotransformers.tests.test_logits.test_logits_type_fasta(init_model, batch_size, fasta_path, lengths_sequence_fasta, pass_mode)