(embedding_size, cross_num, hidden_size, sparse_feature_num)
| 11 | ] # ('auto', 1, (32,), 3) , ('auto', 1, (), 1), ('auto', 1, (32,), 3) |
| 12 | ) |
| 13 | def test_DCNMix(embedding_size, cross_num, hidden_size, sparse_feature_num): |
| 14 | model_name = "DCN-Mix" |
| 15 | |
| 16 | sample_size = SAMPLE_SIZE |
| 17 | x, y, feature_columns = get_test_data( |
| 18 | sample_size, sparse_feature_num=sparse_feature_num, dense_feature_num=sparse_feature_num) |
| 19 | |
| 20 | model = DCNMix(linear_feature_columns=feature_columns, dnn_feature_columns=feature_columns, |
| 21 | cross_num=cross_num, dnn_hidden_units=hidden_size, dnn_dropout=0.5, device=get_device()) |
| 22 | check_model(model, model_name, x, y) |
| 23 | |
| 24 | |
| 25 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected