From dbfedefa40aa9552ccf68a95c1c088324a084499 Mon Sep 17 00:00:00 2001 From: yangyaming Date: Wed, 15 Nov 2017 18:09:45 +0800 Subject: [PATCH] Bug fix for ctr model. --- ctr/avazu_data_processer.py | 1 + ctr/network_conf.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/ctr/avazu_data_processer.py b/ctr/avazu_data_processer.py index dd148adc24..18aa85330a 100644 --- a/ctr/avazu_data_processer.py +++ b/ctr/avazu_data_processer.py @@ -2,6 +2,7 @@ import csv import cPickle import argparse +import os import numpy as np from utils import logger, TaskMode diff --git a/ctr/network_conf.py b/ctr/network_conf.py index b01e4872f1..f6f4e4a59e 100644 --- a/ctr/network_conf.py +++ b/ctr/network_conf.py @@ -50,7 +50,7 @@ def _declare_input_layers(self): 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(