When I try to run the file in the title, I get tensorflow.python.framework,errors_impl.InvalidArgumentError: Expect 4 fields but have 2 in record [Op:IteratorGetNext].
Stack trace is as follows:
File "/home/gregory.werner/eclipse-workspace/SemiSupervisedBertNer/Train1.py", line 13, in
model = trainer.train(labelled_ds=labelled_ds, test_ds=test_ds, unlabelled_ds=unlabelled_ds, config)
File "/home/gregory.werner/eclipse-workspace/SemiSupervisedBertNer/ner/trainers/bert_ner_trainer_confidence_kl.py", line 120, in train
trained_model = self.train_loop(model,ner_optimizer, kl_optimizer, kwargs)
File "/home/gregory.werner/eclipse-workspace/SemiSupervisedBertNer/ner/trainers/bert_ner_trainer_confidence_kl.py", line 70, in train_loop
for x in unlabelled_ds.take(1):
File "/home/gregory.werner/.local/lib/python3.6/site-packages/tensorflow/python/data/ops/iterator_ops.py", line 761, in next
return self._next_internal()
File "/home/gregory.werner/.local/lib/python3.6/site-packages/tensorflow/python/data/ops/iterator_ops.py", line 747, in _next_internal
output_shapes=self._flat_output_shapes)
File "/home/gregory.werner/.local/lib/python3.6/site-packages/tensorflow/python/ops/gen_dataset_ops.py", line 2728, in iterator_get_next
_ops.raise_from_not_ok_status(e, name)
File "/home/gregory.werner/.local/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 6897, in raise_from_not_ok_status
six.raise_from(core._status_to_exception(e.code, message), None)
File "", line 3, in raise_from
When I try to run the file in the title, I get
tensorflow.python.framework,errors_impl.InvalidArgumentError: Expect 4 fields but have 2 in record [Op:IteratorGetNext]
.Stack trace is as follows: