Skip to content

Commit 7d5bfd9

Browse files
author
浅梦
authored
Update CCPM_test.py
1 parent 33faa4b commit 7d5bfd9

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/models/CCPM_test.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def test_CCPM(sparse_feature_num, dense_feature_num):
1616
x, y, feature_dim_dict = get_test_data(
1717
sample_size, sparse_feature_num, dense_feature_num)
1818

19-
model = CCPM(feature_dim_dict, filter_width=(3, 2), feature_maps=(2, 1), hidden_size=[32, ], keep_prob=0.5, )
19+
model = CCPM(feature_dim_dict, conv_kernel_width=(3, 2), conv_filters=(2, 1), hidden_size=[32, ], keep_prob=0.5, )
2020
check_model(model, model_name, x, y)
2121

2222

@@ -32,10 +32,9 @@ def test_CCPM_without_seq(sparse_feature_num, dense_feature_num):
3232
x, y, feature_dim_dict = get_test_data(
3333
sample_size, sparse_feature_num, dense_feature_num, sequence_feature=())
3434

35-
model = CCPM(feature_dim_dict, filter_width=(3, 2), feature_maps=(2, 1), hidden_size=[32, ], keep_prob=0.5, )
35+
model = CCPM(feature_dim_dict, conv_kernel_width=(3, 2), conv_filters=(2, 1), hidden_size=[32, ], keep_prob=0.5, )
3636
check_model(model, model_name, x, y)
3737

3838

3939
if __name__ == "__main__":
40-
# CCPM(2, 2)
4140
pass

0 commit comments

Comments
 (0)