Contents

biotransformers.tests.test_logits

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)