AdamStein97 / Semi-Supervised-BERT-NER

31 stars 8 forks source link

Running example file example_train_bert_confidence_kl.py fails #8

Closed demongolem-biz closed 3 years ago

demongolem-biz commented 3 years ago

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

demongolem-biz commented 3 years ago

CSV was corrupted during download. Closing.