提交 dbfedefa 编写于 作者: Y yangyaming

Bug fix for ctr model.

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