def test_model(classify_df, classify_df_atomizer):
"""Run common model tests."""
testlib.HeterogeneousMappingModelTest(
deeptune.DeepTune,
classify_df,
classify_df_atomizer,
{
"lstm_layer_size": 8,
"dense_layer_size": 4,
"num_epochs": 2,
"batch_size": 4,
"input_shape": (10, ),
},
)
def test_model(classify_df, classify_df_atomizer):
"""Run common model tests."""
testlib.HeterogeneousMappingModelTest(static_mapping.StaticMapping,
classify_df, classify_df_atomizer,
{})
def test_model(classify_df, classify_df_atomizer):
"""Run common model tests."""
testlib.HeterogeneousMappingModelTest(lda.Lda, classify_df,
classify_df_atomizer, {})