fix

Author: yinhaofeng

models/multitask/mmoe/census_reader.py

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

models/rank/logistic_regression/criteo_lr_reader.py

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

models/rank/logistic_regression/dygraph_model.py

@@ -35,7 +35,7 @@ def create_model(self, config):
     # define feeds which convert numpy of batch data to paddle.tensor 
     def create_feeds(self, batch_data, config):
         num_field = config.get('hyper_parameters.num_field', None)
-        label = paddle.to_tensor(batch_data[0].numpy().astype('int32').reshape(
+        label = paddle.to_tensor(batch_data[0].numpy().astype('int64').reshape(
             -1, 1))
         feat_idx = paddle.to_tensor(batch_data[1].numpy().astype('int64')
                                     .reshape(-1, 1))