Skip to content

Commit 2e8cf80

Browse files
authored
Merge pull request tensorflow#8183 from jasonmorton/patch-1
Correct shape->dense_shape in SparseTensor
2 parents e05e5ed + ae1668f commit 2e8cf80

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow/docs_src/tutorials/wide_and_deep.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ def input_fn(df):
233233
categorical_cols = {k: tf.SparseTensor(
234234
indices=[[i, 0] for i in range(df[k].size)],
235235
values=df[k].values,
236-
shape=[df[k].size, 1])
236+
dense_shape=[df[k].size, 1])
237237
for k in CATEGORICAL_COLUMNS}
238238
# Merges the two dictionaries into one.
239239
feature_cols = dict(continuous_cols.items() + categorical_cols.items())

0 commit comments

Comments
 (0)