Skip to content

Commit aeb8cfc

Browse files
authored
Merge pull request tensorflow#3122 from knowm/master
[blog_estimators_dataset.py] bug fix for failing demo
2 parents f5125dd + 89f5e4e commit aeb8cfc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

samples/outreach/blogs/blog_estimators_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def downloadDataset(url, file):
6565
def my_input_fn(file_path, perform_shuffle=False, repeat_count=1):
6666
def decode_csv(line):
6767
parsed_line = tf.decode_csv(line, [[0.], [0.], [0.], [0.], [0]])
68-
label = parsed_line[-1:] # Last element is the label
68+
label = parsed_line[-1] # Last element is the label
6969
del parsed_line[-1] # Delete last element
7070
features = parsed_line # Everything but last elements are the features
7171
d = dict(zip(feature_names, features)), label

0 commit comments

Comments
 (0)