@@ -54,8 +54,8 @@ def ctr_reader(
54
54
feed_dict ,
55
55
file_type , # gzip or plain
56
56
file_format , # csv or svm
57
- dense_slot_indexs ,
58
- sparse_slot_indexs ,
57
+ dense_slot_index ,
58
+ sparse_slot_index ,
59
59
capacity ,
60
60
thread_num ,
61
61
batch_size ,
@@ -78,11 +78,20 @@ def ctr_reader(
78
78
Note that :code:`Program.clone()` method cannot clone :code:`py_reader`.
79
79
80
80
Args:
81
+ feed_dict(list(variable)): a list of data variable.
82
+ file_type('gzip'|'plain'): the type of the data file
83
+ file_format('csv'|'svm'): csv data or svm data format.
84
+ cvs data format is :
85
+ label dense_fea,dense_fea sparse_fea,sparse_fea
86
+ the svm data format is :
87
+ label slot1:fea_sign slot2:fea_sign slot1:fea_sign
88
+ dense_slot_index(list(int)): the index of dense slots
89
+ sparse_slot_index(list(int)): the index of sparse slots
81
90
capacity(int): The buffer capacity maintained by :code:`py_reader`.
82
91
thread_num(list|tuple): List of tuples which declaring data shapes.
83
92
batch_size(list|tuple): List of strs which declaring data type.
84
93
file_list(list|tuple): List of ints which declaring data lod_level.
85
- slots(bool): Whether use double buffer or not.
94
+ slots(bool): slot id of all sparse feature
86
95
name(basestring): The prefix Python queue name and Reader name. None will
87
96
be generated automatically.
88
97
@@ -116,8 +125,8 @@ def ctr_reader(
116
125
'file_list' : file_list ,
117
126
'file_type' : file_type ,
118
127
'file_format' : file_format ,
119
- 'dense_slot_index' : dense_slot_indexs ,
120
- 'sparse_slot_index' : sparse_slot_indexs ,
128
+ 'dense_slot_index' : dense_slot_index ,
129
+ 'sparse_slot_index' : sparse_slot_index ,
121
130
'sparse_slots' : slots ,
122
131
'ranks' : [],
123
132
'lod_levels' : [],
0 commit comments