diff --git a/ctr/avazu_data_processer.py b/ctr/avazu_data_processer.py index dd148adc244efc64021446b17488ec7f2b1c9bd9..18aa85330ae765a236416ab1826092c1fe6a133a 100644 --- a/ctr/avazu_data_processer.py +++ b/ctr/avazu_data_processer.py @@ -2,6 +2,7 @@ import sys 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 b01e4872f108896f60f5d662fe6e1d57295de3f2..f6f4e4a59e17114b279149048c3bf144f2987faa 100644 --- a/ctr/network_conf.py +++ b/ctr/network_conf.py @@ -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(