test.examples.gsf.mltr30k package

Submodules

test.examples.gsf.mltr30k.test_dataset module

Documentation

class test.examples.gsf.mltr30k.test_dataset.TestDataset(methodName='runTest')

Bases: unittest.case.TestCase

Create an instance of the class that will use the named test method when executed. Raises a ValueError if the instance does not have a method with the specified name.

data_file = 'data/mltr30k/valid.tfrecords.gz'

dataset = None

classmethod setUpClass() → None

Hook method for setting up class fixture before running tests in the class.

stats_file = 'data/mltr30k/train.stats.pkl'

test_dataset()

test_sample_documents_flat_indices()

test_sample_windowed()

test_shuffle_documents()

test_truncate_list_size()

test.examples.gsf.mltr30k.test_dataset.pairwise(iterable)

test.examples.gsf.mltr30k.test_model module

Documentation

class test.examples.gsf.mltr30k.test_model.MyTestCase(methodName='runTest')

Bases: unittest.case.TestCase

Create an instance of the class that will use the named test method when executed. Raises a ValueError if the instance does not have a method with the specified name.

group_size = 4

multiple = 2

padded_shapes = ({'context_one_hot': {}, 'context_multi_hot': {}, 'context_dense': {}, 'sequence_one_hot': {}, 'sequence_multi_hot': {}, 'sequence_dense': (None, None)}, (None,))

padding_values = ({'context_one_hot': 0.0, 'context_multi_hot': 0.0, 'context_dense': 0.0, 'sequence_one_hot': 0.0, 'sequence_multi_hot': 0.0, 'sequence_dense': 0.0}, -3.4028235e+38)

seed = 0

classmethod setUpClass() → None

Hook method for setting up class fixture before running tests in the class.

test_gin_forward_pass()

test_gsf_foward_pass()

Module contents

Documentation