Function test_FiBiNET

tests/models/FiBiNET_test.py:13–20

(bilinear_type)

     "interaction", "all"]
)
def test_FiBiNET(bilinear_type):
    model_name = "FiBiNET"
    sample_size = SAMPLE_SIZE
    x, y, feature_columns = get_test_data(sample_size, sparse_feature_num=3, dense_feature_num=3)

    model = FiBiNET(feature_columns, feature_columns,
                    bilinear_type=bilinear_type, dnn_hidden_units=[8, 8], dnn_dropout=0.5, device=get_device())
    check_model(model, model_name, x, y)


if __name__ == "__main__":