We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 856a949 commit a5c6012Copy full SHA for a5c6012
tests/data/huggingface/run_tf.py
@@ -65,8 +65,7 @@
65
x: train_dataset[x].to_tensor(default_value=0, shape=[None, tokenizer.model_max_length])
66
for x in ["input_ids", "attention_mask"]
67
}
68
-
69
+
70
tf_train_dataset = tf.data.Dataset.from_tensor_slices(
71
(train_features, train_dataset["label"])
72
).batch(args.per_device_train_batch_size)
@@ -85,7 +84,6 @@
85
84
86
87
88
89
tf_test_dataset = tf.data.Dataset.from_tensor_slices(
90
(test_features, test_dataset["label"])
91
).batch(args.per_device_eval_batch_size)
0 commit comments