Skip to content

Commit ae1668f

Browse files
authored
Correct shape->dense_shape in SparseTensor
1 parent e05e5ed commit ae1668f

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)