(afn_dnn_hidden_units, sparse_feature_num, dense_feature_num)
| 12 | ((32, 16), 0, 3)] |
| 13 | ) |
| 14 | def test_AFN(afn_dnn_hidden_units, sparse_feature_num, dense_feature_num): |
| 15 | model_name = 'AFN' |
| 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=dense_feature_num) |
| 19 | |
| 20 | model = AFN(feature_columns, feature_columns, afn_dnn_hidden_units=afn_dnn_hidden_units, device=get_device()) |
| 21 | |
| 22 | check_model(model, model_name, x, y) |
| 23 | |
| 24 | |
| 25 | if __name__ == '__main__': |
nothing calls this directly
no test coverage detected