We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents f5125dd + 89f5e4e commit aeb8cfcCopy full SHA for aeb8cfc
samples/outreach/blogs/blog_estimators_dataset.py
@@ -65,7 +65,7 @@ def downloadDataset(url, file):
65
def my_input_fn(file_path, perform_shuffle=False, repeat_count=1):
66
def decode_csv(line):
67
parsed_line = tf.decode_csv(line, [[0.], [0.], [0.], [0.], [0]])
68
- label = parsed_line[-1:] # Last element is the label
+ label = parsed_line[-1] # Last element is the label
69
del parsed_line[-1] # Delete last element
70
features = parsed_line # Everything but last elements are the features
71
d = dict(zip(feature_names, features)), label
0 commit comments