提交 dbfedefa 编写于 作者: Y yangyaming

Bug fix for ctr model.

上级 616c2d98
......@@ -2,6 +2,7 @@ import sys
import csv
import cPickle
import argparse
import os
import numpy as np
from utils import logger, TaskMode
......
......@@ -50,7 +50,7 @@ class CTRmodel(object):
self.lr_merged_input = layer.data(
name='lr_input',
type=paddle.data_type.sparse_vector(self.lr_input_dim))
type=paddle.data_type.sparse_float_vector(self.lr_input_dim))
if not self.is_infer:
self.click = paddle.layer.data(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册