experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix another bug in computing shape for SDCA fake bias column

Browse Source 
Change: 146266321
This commit is contained in:
David Soergel 2017-02-01 11:16:28 -08:00 committed by TensorFlower Gardener
parent 667bf9879f
commit 287e845c52
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tensorflow/contrib/learn/python/learn/estimators/linear.py
View File

@ -78,12 +78,21 @@ def _add_bias_column(feature_columns, columns_to_tensors, bias_variable,
if not feature_columns: if not feature_columns:
raise ValueError("feature_columns can't be empty.") raise ValueError("feature_columns can't be empty.")
# Using an arbitrary input tensor to figure out batch_size. # Loop through input tensors until we can figure out batch_size.
some_input = next(iter(columns_to_tensors.values())) batch_size = None
if isinstance(some_input, sparse_tensor.SparseTensor): for column in columns_to_tensors.values():
batch_size = tensor_util.constant_value(some_input.dense_shape)[0] if isinstance(column, tuple):
column = column[0]
if isinstance(column, sparse_tensor.SparseTensor):
shape = tensor_util.constant_value(column.dense_shape)
if shape is not None:
batch_size = shape[0]
break
else: else:
batch_size = array_ops.shape(some_input)[0] batch_size = array_ops.shape(column)[0]
break
if batch_size is None:
raise ValueError("Could not infer batch size from input features.")
bias_column = layers.real_valued_column(bias_column_name) bias_column = layers.real_valued_column(bias_column_name)
columns_to_tensors[bias_column] = array_ops.ones([batch_size, 1], columns_to_tensors[bias_column] = array_ops.ones([batch_size, 1],