@@ -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
3939if __name__ == "__main__" :
40- # CCPM(2, 2)
4140 pass
0 commit comments