class TestBase[source]

TestBase()

class PysparkTestBase[source]

PysparkTestBase() :: TestBase

create_dummy_features_hidden_features[source]

create_dummy_features_hidden_features(batch_size=1, hidden_dim=768, sample_features:dict=None, problem:str=None)

test_top_layer[source]

test_top_layer(top_class, problem:str, params:Params, sample_features:dict, hidden_dim:int, test_batch_size_list:list=None, **kwargs)