fix

Author: yinhaofeng

models/multitask/esmm/esmm_reader.py

@@ -64,12 +64,14 @@ def __iter__(self):
                         self.all_field_id_dict[field_id][0] = False
                         if len(output[index][1]) > self.max_len:
                             output_list.append(
-                                np.array(output[index][1][:self.max_len]))
+                                np.array(output[index][1][:self.max_len])
+                                .astype('int64'))
                         else:
                             for ii in range(self.max_len - len(output[index][
                                     1])):
                                 output[index][1].append(self.padding)
-                            output_list.append(np.array(output[index][1]))
-                    output_list.append(np.array([ctr]))
-                    output_list.append(np.array([ctcvr]))
+                            output_list.append(
+                                np.array(output[index][1]).astype('int64'))
+                    output_list.append(np.array([ctr]).astype('int64'))
+                    output_list.append(np.array([ctcvr]).astype('int64'))
                     yield output_list

models/multitask/mmoe/census_reader.py

@@ -45,6 +45,8 @@ def __iter__(self):
                         label_marital = [1]
                     output_list = []
                     output_list.append(np.array(data).astype('float32'))
-                    output_list.append(np.array(label_income))
-                    output_list.append(np.array(label_marital))
+                    output_list.append(
+                        np.array(label_income).astype('float32'))
+                    output_list.append(
+                        np.array(label_marital).astype('float32'))
                     yield output_list

models/rank/deepfm/criteo_reader.py

@@ -73,7 +73,7 @@ def __iter__(self):
                     # sparse
                     output_list = []
                     for key, value in output[:-1]:
-                        output_list.append(np.array(value))
+                        output_list.append(np.array(value).astype('int64'))
                     # dense
                     output_list.append(
                         np.array(output[-1][1]).astype("float32"))

models/rank/dnn/criteo_reader.py

@@ -73,7 +73,7 @@ def __iter__(self):
                     # sparse
                     output_list = []
                     for key, value in output[:-1]:
-                        output_list.append(np.array(value))
+                        output_list.append(np.array(value).astype('int64'))
                     # dense
                     output_list.append(
                         np.array(output[-1][1]).astype("float32"))

models/rank/fm/criteo_reader.py

@@ -73,7 +73,7 @@ def __iter__(self):
                     # sparse
                     output_list = []
                     for key, value in output[:-1]:
-                        output_list.append(np.array(value))
+                        output_list.append(np.array(value).astype('int64'))
                     # dense
                     output_list.append(
                         np.array(output[-1][1]).astype("float32"))

models/rank/logistic_regression/criteo_lr_reader.py

@@ -70,5 +70,6 @@ def __iter__(self):
                         else:
                             self.visit[slot] = False
                     # label, feat_idx, feat_value
-                    yield np.array(output[0][1]), np.array(output[1][
-                        1]), np.array(output[2][1]).astype('float32')
+                    yield np.array(output[0][1]).astype('int32'), np.array(
+                        output[1][1]).astype('int64'), np.array(output[2][
+                            1]).astype('float32')

models/rank/wide_deep/criteo_reader.py

@@ -72,7 +72,7 @@ def __iter__(self):
                     # sparse
                     output_list = []
                     for key, value in output[:-1]:
-                        output_list.append(np.array(value))
+                        output_list.append(np.array(value).astype('int64'))
                     # dense
                     output_list.append(
                         np.array(output[-1][1]).astype("float32"))

models/recall/word2vec/word2vec_infer_reader.py

@@ -90,9 +90,10 @@ def __iter__(self):
                     output_list = []
                     for i in range(4):
                         output_list.append(
-                            np.array([self.word_to_id[features[i]]]))
+                            np.array([self.word_to_id[features[i]]]).astype(
+                                'int64'))
                     inputs_words = [
                         self.word_to_id[features[i]] for i in range(3)
                     ]
-                    output_list.append(np.array(inputs_words))
+                    output_list.append(np.array(inputs_words).astype('int64'))
                     yield output_list

models/recall/word2vec/word2vec_reader.py

@@ -176,9 +176,10 @@ def __iter__(self):
                     output_list = []
                     for i in range(4):
                         output_list.append(
-                            np.array([self.word_to_id[features[i]]]))
+                            np.array([self.word_to_id[features[i]]]).astype(
+                                'int64'))
                     inputs_words = [
                         self.word_to_id[features[i]] for i in range(3)
                     ]
-                    output_list.append(np.array(inputs_words))
+                    output_list.append(np.array(inputs_words).astype('int64'))
                     yield output_list